Add slots tests for Vertex and Edge

realshouzy · Apr 28, 2024 · dec0729 · dec0729
1 parent f75af4f
commit dec0729
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tests/datastructures/edge_test.py b/tests/datastructures/edge_test.py
@@ -8,6 +8,10 @@
 from nrw.datastructures._vertex import Vertex
 
 
+def test_slots_of_edge() -> None:
+    assert Edge.__slots__ == ("_vertices", "_weight", "_mark")
+
+
 def test_edge_is_unhashable() -> None:
     assert Edge.__hash__ is None
 

diff --git a/tests/datastructures/vertex_test.py b/tests/datastructures/vertex_test.py
@@ -7,6 +7,10 @@
 from nrw.datastructures._vertex import Vertex
 
 
+def test_slots_of_vertex() -> None:
+    assert Vertex.__slots__ == ("_id", "_mark")
+
+
 def test_vertex_is_unhashable() -> None:
     assert Vertex.__hash__ is None