Enhance __setitem__ and __getitem__ methods to support Ellipsis index…

…ing in CasadiLike class

src/adam/casadi/casadi_like.py

@@ -85,11 +85,27 @@ def __truediv__(self, other: Union["CasadiLike", npt.ArrayLike]) -> "CasadiLike"
 
     def __setitem__(self, idx, value: Union["CasadiLike", npt.ArrayLike]):
         """Overrides set item operator"""
-        self.array[idx] = value.array if type(self) is type(value) else value
+        if idx is Ellipsis:
+            self.array = value.array if isinstance(value, CasadiLike) else value
+        elif isinstance(idx, tuple) and Ellipsis in idx:
+            idx = tuple(slice(None) if i is Ellipsis else i for i in idx)
+            self.array[idx] = value.array if isinstance(value, CasadiLike) else value
+        else:
+            self.array[idx] = value.array if isinstance(value, CasadiLike) else value
+
 
     def __getitem__(self, idx) -> "CasadiLike":
         """Overrides get item operator"""
-        return CasadiLike(self.array[idx])
+        if idx is Ellipsis:
+            # Handle the case where only Ellipsis is passed
+            return CasadiLike(self.array)
+        elif isinstance(idx, tuple) and Ellipsis in idx:
+            # Handle the case where Ellipsis is part of a tuple
+            idx = tuple(slice(None) if k is Ellipsis else k for k in idx)
+            return CasadiLike(self.array[idx])
+        else:
+            # For other cases, delegate to the CasADi object's __getitem__
+            return CasadiLike(self.array[idx])
 
     @property
     def T(self) -> "CasadiLike":