add compatibility with the numpy roll and DWA objects

src/pymodaq_data/numpy_func.py

@@ -146,6 +146,12 @@ def _abs(dwa: 'DataWithAxes', *args, **kwargs):
     return np.absolute(dwa, *args, **kwargs)
 
 
+@implements('roll')
+def _roll(dwa: 'DataWithAxes', *args, **kwargs):
+    dwa_func = dwa.deepcopy_with_new_data(data=[np.roll(array, *args, **kwargs) for array in dwa])
+    dwa_func.name += f"_{'roll'}"
+    return dwa_func
+
 # ******** functions that return booleans ***********
 @implements('all')
 def _all(dwa: 'DataWithAxes', *args, axis: Optional[Union[int, Iterable[int]]] = None, **kwargs):

tests/data_test.py

@@ -1593,4 +1593,8 @@ def test_flip_transpose_rotate(self):
         for ind in range(len(dwa)):
             assert np.allclose(np.transpose(dwa[ind]), dwa_transform[ind])
 
-
+    def test_roll(self):
+        dwa = data_mod.DataRaw('raw', units='', data=[DATA2D])
+        SHIFT = (10, 5)
+        dwa_rolled = np.roll(dwa, shift = SHIFT)
+        assert np.allclose(dwa_rolled[0], np.roll(dwa[0], shift=SHIFT))