Use 1D size for random array creation

jni · jni · commit 2e4ee62338f3 · 2019-10-15T09:06:22.000-05:00
diff --git a/test/test_array.py b/test/test_array.py
@@ -667,7 +667,7 @@ def test_random_int_in_range(ctx_factory, rng_class, dtype, plot_hist=False):
 def test_array_method(ctx_factory):
     ctx = ctx_factory()
     queue = cl.CommandQueue(ctx)
-    arr = make_random_array(queue, np.float32, (32, 32))
+    arr = make_random_array(queue, np.float32, 1024)
     np.testing.assert_array_equal(arr, arr.get())
 
 def test_numpy_integer_shape(ctx_factory):
