fix: test also lazy arrays

Author: paquiteau

src/mrinufft/_array_compat.py

@@ -321,6 +321,13 @@ def _convert(_array_to_xp, args, kwargs=None):
         elif isinstance(_arg, (tuple, list)):
             args[n], _ = _convert(_array_to_xp, _arg)
         # objects with attributes that are arrays are also converted
+        elif hasattr(_arg, "__dict__") and not isinstance:
+            process_dict_vals, _ = _convert(*_arg.__dict__.values())
+            for k, v in zip(_arg.__dict__.keys(), process_dict_vals):
+                try:
+                    setattr(_arg, k, v)
+                except Exception:
+                    pass
     # convert keyworded
     if kwargs:
         process_kwargs_vals, _ = _convert(_array_to_xp, kwargs.values())

src/mrinufft/extras/field_map.py

@@ -270,10 +270,6 @@ def __len__(self) -> int:
         """Get number of interpolators."""
         return len(self.C_small)
 
-    def __getattr__(self, name):
-        """Get other attribute from array."""
-        return getattr(self.C_small, name)
-
     @property
     def shape(self):
         """Overall shape of the lazy array."""

tests/test_offres_exp_approx.py

@@ -60,8 +60,8 @@ def case_complex2D(self, N=64, b0range=(-300, 300), t2svalue=15.0):
 @_param_array_interface_np_cp
 @parametrize_with_cases("b0_map, r2s_map, mask", cases=CasesB0maps)
 @parametrize("method", ["svd-full", "mti", "mfi"])
-@parametrize("L", [40, -1])
-def test_b0map_coeff(b0_map, r2s_map, mask, method, L, array_interface):
+@parametrize("L, lazy", [(40, True), (-1, True), (40, False)])
+def test_b0map_coeff(b0_map, r2s_map, mask, method, L, lazy, array_interface):
     """Test exponential approximation for B0 field only."""
     # Generate readout times
     Nt = 400
@@ -80,7 +80,7 @@ def test_b0map_coeff(b0_map, r2s_map, mask, method, L, array_interface):
         to_interface(tread, array_interface),
         to_interface(mask, array_interface),
         L=L,
-        lazy=False,
+        lazy=lazy,
         n_bins=4096,
         **kwargs,
     )
@@ -92,6 +92,8 @@ def test_b0map_coeff(b0_map, r2s_map, mask, method, L, array_interface):
     assert B.shape == (Nt, L)
     assert C.shape == (L, *b0_map.shape)
 
+    if lazy:
+        C = np.stack([C[l] for l in range(len(C))])
     # Check that the approximation match the full matrix.
     B = from_interface(B, array_interface)
     C = from_interface(C, array_interface)