chore: add methods for making D2Connection hashable and comparable …

…to easily ensure uniqueness if needed (#17)

chore: add methods for making `D2Connection` hashable and comparable to easily ensure uniqueness if needed

src/py_d2/connection.py

@@ -26,3 +26,11 @@ def lines(self) -> List[str]:
 
  def __repr__(self) -> str:
  return "\n".join(self.lines())
+
+ def __hash__(self):
+ return hash((self.shape_1, self.shape_2, self.label, self.direction))
+
+ def __eq__(self, other) -> bool:
+ if ((self.shape_1, self.shape_2, self.direction, self.label) == (other.shape_1, other.shape_2, other.direction, other.label)):
+ return True
+ return False

tests/test_py_d2/test_d2_connection.py

@@ -31,3 +31,9 @@ def test_d2_connection_direction_both():
 def test_d2_connection_direction_none():
  connection = D2Connection(shape_1="a", shape_2="b", direction=Direction.NONE)
  assert str(connection) == "a -- b"
+
+
+def test_d2_connection_uniqueness():
+ connection = D2Connection(shape_1="a", shape_2="b", direction=Direction.TO)
+ connection2 = D2Connection(shape_1="a", shape_2="b", direction=Direction.TO)
+ assert connection == connection2