improved test xarray-contrib#178

feefladder · Apr 6, 2021 · 803f652 · 803f652
1 parent 2173b87
commit 803f652
Showing 1 changed file with 3 additions and 7 deletions.
diff --git 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")