diff --git a/xsimlab/tests/test_model.py b/xsimlab/tests/test_model.py index c087e0c..103a596 100644 --- a/xsimlab/tests/test_model.py +++ b/xsimlab/tests/test_model.py @@ -498,16 +498,12 @@ class C: class D: pass - @xs.process - class E: - pass - model = xs.Model( - {"a": A, "b": B, "c": C, "d": D, "e": E}, - custom_dependencies={"d": "c", "c": "b", "b": {"a", "e"}}, + {"a": A, "b": B, "c": C, "d": D}, + custom_dependencies={"d": "c", "c": "b", "b": "a"}, ) model = model.drop_processes(["b", "c"]) - assert set(model.dependent_processes["d"]) == {"a", "e"} + assert model.dependent_processes["d"] == ["a"] def test_visualize(self, model): pytest.importorskip("graphviz")