Use Parameter.hasCategory in ParameterCollection.where tests

terrapower · Sep 20, 2024 · 83e126f · 83e126f
1 parent 3bb13f7
commit 83e126f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/armi/reactor/tests/test_parameters.py b/armi/reactor/tests/test_parameters.py
@@ -558,7 +558,7 @@ def setUpClass(cls) -> None:
     def test_onCategory(self):
         names = {"keff", "cornerFlux"}
         for p in self.pc.where(
-            lambda pd: parameters.Category.neutronics in pd.categories
+            lambda pd: pd.hasCategory(parameters.Category.neutronics)
         ):
             names.remove(p.name)
         self.assertFalse(names, msg=f"{names=} should be empty!")
@@ -579,7 +579,7 @@ def test_complicated(self):
         }
         for p in self.pc.where(
             lambda pd: pd.atLocation(parameters.ParamLocation.CORNERS)
-            and parameters.Category.neutronics in pd.categories
+            and pd.hasCategory(parameters.Category.neutronics)
         ):
             names.remove(p.name)
         self.assertFalse(names, msg=f"{names=} should be empty")