Let Node(label=name) == name

oemof · Nov 8, 2023 · a765742 · a765742
1 parent aa66673
commit a765742
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/oemof/network/network/entity.py b/src/oemof/network/network/entity.py
@@ -71,7 +71,10 @@ def __init__(self, label=None, *, custom_properties=None, **kwargs):
         self._outputs = Outputs(self)
 
     def __eq__(self, other):
-        return id(self) == id(other)
+        try:
+            return self.label == other.label
+        except AttributeError:
+            return self.label == other
 
     def __lt__(self, other):
         return str(self) < str(other)

diff --git a/tests/test_network_classes.py b/tests/test_network_classes.py
@@ -372,5 +372,18 @@ def test_comparision():
     node1 = Node(label=2)
     node2 = Node(label=-5)
 
+    assert node0 == 0
+    assert 2 == node1
     assert node0 < node1
     assert node0 > node2
+
+
+def test_dict_access():
+    node0 = Node(label="label_0")
+    test_dict = {node0: 5}
+    assert test_dict[node0] == 5
+    assert test_dict["label_0"] == 5
+
+    test_dict = {"label_0": 5}
+    assert test_dict[node0] == 5
+    assert test_dict["label_0"] == 5