From a7dbe0a91f13f644a918a1df1cb23a5664c03f0c Mon Sep 17 00:00:00 2001 From: Ewout ter Hoeven Date: Thu, 3 Oct 2024 15:16:16 +0200 Subject: [PATCH] Fix test --- tests/test_solara_viz.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_solara_viz.py b/tests/test_solara_viz.py index 301294f25ba..a0d2b449399 100644 --- a/tests/test_solara_viz.py +++ b/tests/test_solara_viz.py @@ -100,11 +100,14 @@ 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) + mock_space_matplotlib.assert_called_with( + model, agent_portrayal, propertylayer_portrayal + ) # specify no space should be drawn mock_space_matplotlib.reset_mock()