Raise error on slicing a dt

Closes #57
rjfarmer · Feb 29, 2024 · da40381 · da40381
1 parent ddc5227
commit da40381
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/gfort2py/fDT.py b/gfort2py/fDT.py
@@ -136,12 +136,18 @@ def __contains__(self, key):
         return key in self.keys()
 
     def __getitem__(self, key):
+        if isinstance(key, slice):
+            raise AttributeError("Scalar derived type can't be sliced")
+
         if key in self._dt_args:
             return self._dt_args[key].from_ctype(getattr(self.cvalue, key))
         else:
             raise KeyError(f"{key} not present in {self._dt_obj.name}")
 
     def __setitem__(self, key, value):
+        if isinstance(key, slice):
+            raise AttributeError("Scalar derived type can't be sliced")
+
         if key in self._dt_args:
             self.from_param({key: value})
         else: