Test that no flows are removed when adding new

oemof · Sep 29, 2023 · fdfac06 · fdfac06
1 parent c691376
commit fdfac06
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/test_energy_system.py b/tests/test_energy_system.py
@@ -45,17 +45,20 @@ def test_add_nodes(self):
     def test_add_flow(self):
         assert not self.es.nodes
 
+        node0 = Node(label="node0")
         node1 = Node(label="node1")
-        node2 = Node(label="node2")
-
-        self.es.add(node1, node2)
+        node2 = Node(label="node2", inputs={node0: Edge()})
 
+        self.es.add(node0, node1, node2)
         node2.add_inputs({node1: Edge()})
 
+        assert (node0, node2) in self.es.flows().keys()
         assert (node1, node2) in self.es.flows().keys()
         assert (node2, node1) not in self.es.flows().keys()
 
         node2.add_outputs({node1: Edge()})
+        assert (node0, node2) in self.es.flows().keys()
+        assert (node1, node2) in self.es.flows().keys()
         assert (node2, node1) in self.es.flows().keys()
 
     def test_that_node_additions_are_signalled(self):