diff --git a/armi/reactor/tests/test_parameters.py b/armi/reactor/tests/test_parameters.py index dbefc9e44..1e1953fe6 100644 --- a/armi/reactor/tests/test_parameters.py +++ b/armi/reactor/tests/test_parameters.py @@ -560,6 +560,7 @@ def test_onCategory(self): for p in self.pc.where( lambda pd: pd.hasCategory(parameters.Category.neutronics) ): + self.assertTrue(p.hasCategory(parameters.Category.neutronics), msg=p) names.remove(p.name) self.assertFalse(names, msg=f"{names=} should be empty!") @@ -570,6 +571,7 @@ def test_onLocation(self): for p in self.pc.where( lambda pd: pd.atLocation(parameters.ParamLocation.EDGES) ): + self.assertTrue(p.atLocation(parameters.ParamLocation.EDGES), msg=p) names.remove(p.name) self.assertFalse(names, msg=f"{names=} should be empty!") @@ -577,9 +579,13 @@ def test_complicated(self): names = { "cornerFlux", } - for p in self.pc.where( - lambda pd: pd.atLocation(parameters.ParamLocation.CORNERS) - and pd.hasCategory(parameters.Category.neutronics) - ): + + def check(p: parameters.Parameter) -> bool: + return p.atLocation(parameters.ParamLocation.CORNERS) and p.hasCategory( + parameters.Category.neutronics + ) + + for p in self.pc.where(check): + self.assertTrue(check(p), msg=p) names.remove(p.name) self.assertFalse(names, msg=f"{names=} should be empty")