Fix test

projectmesa · Oct 3, 2024 · 62be7da · 62be7da
1 parent 77b5711
commit 62be7da
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_solara_viz.py b/tests/test_solara_viz.py
@@ -100,12 +100,13 @@ def test_call_space_drawer(mocker):  # noqa: D103
         "Shape": "circle",
         "color": "gray",
     }
+    propertylayer_portrayal = None
     # initialize with space drawer unspecified (use default)
     # component must be rendered for code to run
     solara.render(SolaraViz(model, components=[make_space_matplotlib(agent_portrayal)]))
     # should call default method with class instance and agent portrayal
     mock_space_matplotlib.assert_called_with(
-        model, agent_portrayal, propertylayer_portrayal=None
+        model, agent_portrayal, propertylayer_portrayal
     )
 
     # specify no space should be drawn