diff --git a/test/test_array.py b/test/test_array.py index e07cedee8d1514f625c2357e34767f0ee556f069..e183a4de0d16aed400bdd7b146f8c85031380921 100644 --- a/test/test_array.py +++ b/test/test_array.py @@ -860,7 +860,9 @@ def test_newaxis(ctx_factory): b = a[:, np.newaxis] assert b_gpu.shape == b.shape - assert b_gpu.strides == b.strides + for i in range(b.ndim): + if b.shape[i] > 1: + assert b_gpu.strides[i] == b.strides[i] if __name__ == "__main__":