Implement contains for strings (uuids)

Author: arneboockmeyer

test/topology_test.py

@@ -126,8 +126,10 @@ def test_contains_method():
     topology.add_signals([signal_1, signal_2])
     topology.add_routes([route_1])
     assert node_1 in topology
+    assert node_1.uuid in topology
     assert node_2 in topology
     assert [node_3, node_4] in topology
+    assert [node_3.uuid, node_4.uuid] in topology
     assert edge_1 in topology
     assert [edge_2, edge_3] in topology
     assert signal_1 in topology

yaramo/topology.py

@@ -65,6 +65,13 @@ def __contains__(self, item_list):
                 all_in = all_in and item in self.signals.values()
             elif type(item) == Route:
                 all_in = all_in and item in self.routes.values()
+            elif type(item) == str:
+                all_in = (
+                    item in self.nodes
+                    or item in self.edges
+                    or item in self.signals
+                    or item in self.routes
+                )
             else:
                 all_in = False
         return all_in