Accept different strides on numpy for newaxis

inducer · Jul 11, 2015 · 955ef2f · 955ef2f
1 parent 68c1fd0
commit 955ef2f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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__":