diff --git a/signal_design/axis.py b/signal_design/axis.py index 3d3cd3e..4dd0ae2 100644 --- 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):