From 38f6029aff6c7bda925ea0b652410e784d4ed7cf Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 24 Apr 2011 02:57:37 -0400
Subject: [PATCH] Speed up nan_arithmetic test.
---
test/test_array.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/test/test_array.py b/test/test_array.py
index 2e71bbda..6be8366c 100644
--- a/test/test_array.py
+++ b/test/test_array.py
@@ -289,8 +289,7 @@ def test_nan_arithmetic(ctx_getter):
ab = a*b
ab_gpu = (a_gpu*b_gpu).get()
- for i in range(size):
- assert np.isnan(ab[i]) == np.isnan(ab_gpu[i])
+ assert (np.isnan(ab) == np.isnan(ab_gpu)).all()
--
GitLab