fix LA[:] = value

gdementen · gdementen · commit cf46c436b577 · 2015-01-28T14:07:05.000+01:00
diff --git a/larray/core.py b/larray/core.py
@@ -1067,6 +1067,9 @@ def __getslice__(self, i, j):
         # sadly LArray[:] translates to LArray.__getslice__(0, sys.maxsize)
         return self[slice(i, j) if i != 0 or j != sys.maxsize else slice(None)]
 
+    def __setslice__(self, i, j, value):
+        self[slice(i, j) if i != 0 or j != sys.maxsize else slice(None)] = value
+
     def __str__(self):
         if not self.ndim:
             return str(np.asscalar(self))
diff --git a/larray/tests/test_la.py b/larray/tests/test_la.py
@@ -676,6 +676,11 @@ def test_setitem_larray(self):
         raw[..., [0, 4, 8]] = 0
         self._assert_equal_raw(la, raw)
 
+        # 4) using a single slice(None) key
+        la = self.larray.copy()
+        la[:] = 0
+        self._assert_equal_raw(la, np.zeros_like(raw))
+
     def test_setitem_ndarray(self):
         """
         tests LArray.__setitem__(key, value) where value is a raw ndarray.
