fix dummy particulator & builder

open-atmos · Aug 2, 2024 · c24a7d3 · c24a7d3
1 parent 4204892
commit c24a7d3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tests/unit_tests/dummy_environment.py b/tests/unit_tests/dummy_environment.py
@@ -2,8 +2,10 @@
 import numpy as np
 
 from PySDM.impl.mesh import Mesh
+from PySDM.environments.impl import register_environment
 
 
+@register_environment()
 class DummyEnvironment:  # pylint: disable=too-many-instance-attributes
     def __init__(
         self,
@@ -33,8 +35,8 @@ def __init__(
                 self.step_counter = 0
         self.courant_field_data = courant_field_data
 
-    def register(self, particulator):
-        self.particulator = particulator
+    def register(self, *, builder):
+        self.particulator = builder.particulator
         if hasattr(self, "halo"):
             self.pred["water_vapour_mixing_ratio"] = particulator.backend.Storage.empty(
                 self.mesh.n_cell, dtype=float

diff --git a/tests/unit_tests/dummy_particulator.py b/tests/unit_tests/dummy_particulator.py
@@ -11,7 +11,7 @@ def __init__(self, backend_class, n_sd=0, formulae=None, grid=None):
         env = DummyEnvironment(grid=grid)
         Builder.__init__(self, n_sd, backend, env)
         Particulator.__init__(self, n_sd, backend)
-        self.environment = env
+        self.environment = env.instantiate(builder=self)
         self.particulator = self
         self.req_attr_names = ["multiplicity", "cell id"]
         self.attributes = None