diff --git a/tests/test_matrix.py b/tests/test_matrix.py index 9416d2e3d0..c8019b4707 100644 --- a/tests/test_matrix.py +++ b/tests/test_matrix.py @@ -61,8 +61,8 @@ class ArrayLike: def __init__(self, data): self.data = data - def __array__(self, dtype=None, copy=None): - return np.asarray(self.data, dtype=dtype, copy=copy) + def __array__(self, **kwargs): + return np.asarray(self.data, **kwargs) p = mat._HeatMapper(ArrayLike(self.x_norm), **self.default_kws) npt.assert_array_equal(p.plot_data, self.x_norm)