Fabian Peter Hammerle пре 5 година
родитељ
комит
235a5a4c65
2 измењених фајлова са 7 додато и 0 уклоњено
  1. 3 0
      freesurfer_surface/__init__.py
  2. 4 0
      tests/test_triangle.py

+ 3 - 0
freesurfer_surface/__init__.py

@@ -86,6 +86,9 @@ class Triangle:
     def __eq__(self, other: 'Triangle') -> bool:
         return self.vertex_indices == other.vertex_indices
 
+    def __repr__(self) -> str:
+        return 'Triangle(vertex_indices={})'.format(self.vertex_indices)
+
 
 class Label:
 

+ 4 - 0
tests/test_triangle.py

@@ -19,3 +19,7 @@ def test_eq():
     assert not Triangle((0, 1, 2)) == Triangle((0, 1, 4))
     assert not Triangle((0, 1, 2)) == Triangle((0, 4, 2))
     assert not Triangle((0, 1, 2)) == Triangle((4, 1, 2))
+
+
+def test_repr():
+    assert repr(Triangle((0, 1, 2))) == 'Triangle(vertex_indices=(0, 1, 2))'