Overwrite __eq__ dunder

src/adam/casadi/casadi_like.py

@@ -93,6 +93,12 @@ def __getitem__(self, idx) -> "CasadiLike":
         """Overrides get item operator"""
         return CasadiLike(self.array[idx])
 
+    def __eq__(self, other: Union["CasadiLike", npt.ArrayLike]) -> bool:
+        """Overrides == operator"""
+        if type(self) is not type(other):
+            return self.array == other
+        return self.array == other.array
+
     @property
     def T(self) -> "CasadiLike":
         """

src/adam/jax/jax_like.py

@@ -107,6 +107,12 @@ def __neg__(self) -> "JaxLike":
         """Overrides - operator"""
         return JaxLike(-self.array)
 
+    def __eq__(self, other: Union["JaxLike", npt.ArrayLike]) -> bool:
+        """Overrides == operator"""
+        if type(self) is not type(other):
+            return self.array.squeeze() == other.squeeze()
+        return self.array.squeeze() == other.array.squeeze()
+
 
 class JaxLikeFactory(ArrayLikeFactory):
     @staticmethod

src/adam/numpy/numpy_like.py

@@ -106,6 +106,13 @@ def __neg__(self):
         """Overrides - operator"""
         return NumpyLike(-self.array)
 
+    def __eq__(self, other: Union["NumpyLike", npt.ArrayLike]) -> bool:
+        """Overrides == operator"""
+        if type(self) is type(other):
+            return self.array == other.array
+        else:
+            return self.array == other
+
 
 class NumpyLikeFactory(ArrayLikeFactory):
     @staticmethod

src/adam/pytorch/torch_like.py

@@ -112,6 +112,13 @@ def __neg__(self) -> "TorchLike":
         """Overrides - operator"""
         return TorchLike(-self.array)
 
+    def __eq__(self, other: Union["TorchLike", ntp.ArrayLike]) -> bool:
+        """Overrides == operator"""
+        if type(self) is type(other):
+            return self.array == other.array
+        else:
+            return self.array == other
+
 
 class TorchLikeFactory(ArrayLikeFactory):
     @staticmethod