add __expr__ for mag class (#226)

scalableminds · Jul 29, 2020 · 80acf22 · 80acf22
1 parent ba415a5
commit 80acf22
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/wkcuber/mag.py b/wkcuber/mag.py
@@ -48,6 +48,9 @@ def __eq__(self, other):
     def __str__(self):
         return self.to_layer_name()
 
+    def __expr__(self):
+        return f"Mag({self.to_layer_name()})"
+
     def to_layer_name(self):
         x, y, z = self.mag
         if x == y and y == z: