Kaynağa Gözat

added Label.__str__()

Fabian Peter Hammerle 5 yıl önce
ebeveyn
işleme
56328decb1
2 değiştirilmiş dosya ile 16 ekleme ve 0 silme
  1. 4 0
      freesurfer_surface/__init__.py
  2. 12 0
      tests/test_label.py

+ 4 - 0
freesurfer_surface/__init__.py

@@ -85,6 +85,10 @@ class Label:
     def hex_color_code(self) -> str:
         return '#{:02x}{:02x}{:02x}'.format(self.red, self.green, self.blue)
 
+    def __str__(self) -> str:
+        return 'Label(name={}, index={}, color={})'.format(
+            self.name, self.index, self.hex_color_code)
+
 
 class Annotation:
 

+ 12 - 0
tests/test_label.py

@@ -19,6 +19,7 @@ def test_color_code(red, green, blue, transparency, color_code):
     label.transparency = transparency
     assert color_code == label.color_code
 
+
 def test_color_code_unknown():
     label = Label()
     label.index = 0
@@ -47,3 +48,14 @@ def test_hex_color_code(red, green, blue, hex_color_code):
     label.green = green
     label.blue = blue
     assert hex_color_code == label.hex_color_code.lower()
+
+
+def test_str():
+    label = Label()
+    label.index = 24
+    label.name = 'precentral'
+    label.red = 60
+    label.green = 20
+    label.blue = 220
+    label.transparency = 0
+    assert str(label) == 'Label(name=precentral, index=24, color=#3c14dc)'