fixed tests

NOAA-GFDL · Jan 3, 2024 · 139c92c · 139c92c
1 parent b7d0de8
commit 139c92c
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 5 deletions.
diff --git a/physics/tests/savepoint/translate/translate_driver.py b/physics/tests/savepoint/translate/translate_driver.py
@@ -8,7 +8,7 @@
 # but also, driver tests should not be in physics
 from pace.fv3core.testing.translate_fvdynamics import TranslateFVDynamics
 from pace.fv3core.testing.validation import enable_selective_validation
-from pace.physics import PhysicsConfig, PhysicsState
+from pace.physics import PHYSICS_PACKAGES, PhysicsConfig, PhysicsState
 from pace.util.namelist import Namelist
 
 
@@ -43,7 +43,8 @@ def compute_parallel(self, inputs, communicator):
             sizer, backend=self.stencil_config.compilation_config.backend
         )
         physics_state = PhysicsState.init_zeros(
-            quantity_factory=quantity_factory, schemes=["GFS_microphysics"]
+            quantity_factory=quantity_factory,
+            schemes=[PHYSICS_PACKAGES["GFS_microphysics"]],
         )
         tendency_state = TendencyState.init_zeros(
             quantity_factory=quantity_factory,

diff --git a/physics/tests/savepoint/translate/translate_gfs_physics_driver.py b/physics/tests/savepoint/translate/translate_gfs_physics_driver.py
@@ -2,6 +2,7 @@
 
 import pace.dsl.gt4py_utils as utils
 import pace.util as util
+from pace.physics import PHYSICS_PACKAGES
 from pace.physics.stencils.physics import Physics, PhysicsState
 from pace.stencils import update_atmos_state
 from pace.stencils.testing.translate_physics import TranslatePhysicsFortranData2Py
@@ -129,7 +130,7 @@ def compute(self, inputs):
         quantity_factory = util.QuantityFactory.from_backend(
             sizer, self.stencil_factory.backend
         )
-        schemes = ["GFS_microphysics"]
+        schemes = [PHYSICS_PACKAGES["GFS_microphysics"]]
         physics_state = PhysicsState(
             **inputs,
             quantity_factory=quantity_factory,

diff --git a/physics/tests/savepoint/translate/translate_microphysics.py b/physics/tests/savepoint/translate/translate_microphysics.py
@@ -5,6 +5,7 @@
 import pace.dsl.gt4py_utils as utils
 import pace.util
 from pace.dsl.typing import Float
+from pace.physics import PHYSICS_PACKAGES
 from pace.physics.stencils.microphysics import Microphysics
 from pace.physics.stencils.physics import PhysicsState
 from pace.stencils.testing.translate_physics import TranslatePhysicsFortranData2Py
@@ -88,7 +89,7 @@ def compute(self, inputs):
             inputs,
             sizer=sizer,
             quantity_factory=quantity_factory,
-            schemes=["GFS_microphysics"],
+            schemes=[PHYSICS_PACKAGES["GFS_microphysics"]],
         )
         microphysics = Microphysics(
             self.stencil_factory, quantity_factory, self.grid.grid_data, self.namelist

diff --git a/tests/main/physics/test_integration.py b/tests/main/physics/test_integration.py
@@ -73,7 +73,7 @@ def setup_physics():
         physics_config,
     )
     physics_state = pace.physics.PhysicsState.init_zeros(
-        quantity_factory, schemes=["GFS_microphysics"]
+        quantity_factory, schemes=[pace.physics.PHYSICS_PACKAGES["GFS_microphysics"]]
     )
     random = np.random.RandomState(0)
     for field in fields(pace.physics.PhysicsState):