TST: add test for setitem in axis 0

Author: Holger Kohr

pygpu/tests/test_gpu_ndarray.py

@@ -310,12 +310,14 @@ def getitem_none(shp):
         assert numpy.allclose(a_gpu[1:, None], a[1:, None])
 
 
-def test_mapping_setitem_ellipsis():
+def test_mapping_setitem():
     for shp in [(9,), (8, 9), (4, 8, 9), (1, 8, 9)]:
         for dtype in dtypes_all:
             for offseted in [True, False]:
                 yield mapping_setitem_ellipsis, shp, dtype, offseted
                 yield mapping_setitem_ellipsis2, shp, dtype, offseted
+                yield mapping_setitem_firstaxis, shp, dtype, offseted
+
 
 @guard_devsup
 def mapping_setitem_ellipsis(shp, dtype, offseted):
@@ -324,13 +326,24 @@ def mapping_setitem_ellipsis(shp, dtype, offseted):
     a_gpu[...] = 2
     assert numpy.allclose(a, numpy.asarray(a_gpu))
 
+
 @guard_devsup
 def mapping_setitem_ellipsis2(shp, dtype, offseted):
     a, a_gpu = gen_gpuarray(shp, dtype, offseted, ctx=ctx)
     b, b_gpu = gen_gpuarray(shp[1:], dtype, False, ctx=ctx)
     a[:] = b
     a_gpu[:] = b_gpu
-    assert numpy.allclose(a, numpy.asarray(b_gpu))
+    assert numpy.allclose(a, numpy.asarray(a_gpu))
+
+
+@guard_devsup
+def mapping_setitem_firstaxis(shp, dtype, offseted):
+    a, a_gpu = gen_gpuarray(shp, dtype, offseted, ctx=ctx)
+    b, b_gpu = gen_gpuarray(shp[1:], dtype, False, ctx=ctx)
+    a[0] = b
+    a_gpu[0] = b_gpu
+    assert numpy.allclose(a, numpy.asarray(a_gpu))
+
 
 class WriteReadTest(unittest.TestCase):
     def setUp(self):