Merge pull request #121 from mariuzka/mk/fix_dummy_model_error

Mk/fix dummy model error

src/pop2net/creator.py

@@ -31,7 +31,7 @@ def __init__(
         self.model = model
         self.seed = seed
         self.rng = random.Random(seed)
-        self._dummy_model = p2n.Model()
+        self._dummy_model = p2n.Model(parameters=self.model.p)
         self._temp_agent_attrs = ["_P2NTEMP_split_values", "_P2NTEMP_melt_location_weight"]
 
     def _create_dummy_location(self, location_cls) -> p2n.Location:

tests/test_creator/test_parameter_transfer.py

@@ -0,0 +1,26 @@
+import networkx as nx
+
+import pop2net as p2n
+
+
+def test_p_transfer():
+    """Tests whether a MagicLocation can access the parameters p of the model
+    when the Creator is working."""
+
+    class TestLocation(p2n.MagicLocation):
+        def __init__(self, model: p2n.Model) -> None:
+            super().__init__(model)
+            self.nxgraph = nx.cycle_graph(n=self.p["n_agents"])
+
+    class TestModel(p2n.Model):
+        def setup(self):
+            self.creator = p2n.Creator(model=self)
+            self.creator.create_agents(n=self.p["n_agents"])
+            self.creator.create_locations(location_classes=[TestLocation])
+
+    params = {"n_agents": 100}
+
+    model = TestModel(parameters=params)
+    model.run(steps=3)
+
+    assert len(model.agents) == 100