feat: add comparation to axis

Omnivanitate · Feb 29, 2024 · 1c7de72 · 1c7de72
1 parent 5cce0b6
commit 1c7de72
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/signal_design/axis.py b/signal_design/axis.py
@@ -178,7 +178,16 @@ def __str__(self):
         )
 
         return result
-
+
+    def __eq__(self, other: 'Axis'):
+        return ( 
+            self.start == other.start 
+            and self.size == self.size 
+            and self.sample == self.sample
+        )
+
+    def __ne__(self, other: 'Axis'):
+        return not self.__eq__(other)
 
 def _check_size(size: int) -> None:
     if not isinstance(size, int):