From 955ef2f0e975c395ff08f347ec86a2c05abb559b Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 11 Jul 2015 15:27:53 -0500
Subject: [PATCH] Accept different strides on numpy for newaxis
---
test/test_array.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/test/test_array.py b/test/test_array.py
index e07cedee..e183a4de 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__":
--
GitLab