Added test_dims.py to public CI (#1588)

Author: antonwolfy

.github/workflows/conda-package.yml

@@ -32,6 +32,7 @@ env:
       third_party/cupy/logic_tests/test_comparison.py
       third_party/cupy/logic_tests/test_truth.py
       third_party/cupy/manipulation_tests/test_basic.py
+      third_party/cupy/manipulation_tests/test_dims.py
       third_party/cupy/manipulation_tests/test_join.py
       third_party/cupy/manipulation_tests/test_rearrange.py
       third_party/cupy/manipulation_tests/test_transpose.py

tests/third_party/cupy/manipulation_tests/test_dims.py

@@ -7,7 +7,6 @@
 from tests.third_party.cupy import testing
 
 
-@testing.gpu
 class TestDims(unittest.TestCase):
     def check_atleast(self, func, xp):
         a = testing.shaped_arange((), xp)
@@ -55,6 +54,13 @@ def test_broadcast_to(self, xp, dtype):
         b = xp.broadcast_to(a, (2, 3, 3, 4))
         return b
 
+    @testing.numpy_cupy_array_equal()
+    def test_broadcast_to_int(self, xp):
+        # Broadcast 0-dim array to 1-dim.
+        a = xp.array(10, dtype=xp.float32)
+        b = xp.broadcast_to(a, 10)
+        return b
+
     @testing.for_all_dtypes()
     def test_broadcast_to_fail(self, dtype):
         for xp in (numpy, cupy):
@@ -286,7 +292,6 @@ def test_external_squeeze(self, xp):
     {"shapes": [(0, 1, 1, 3), (2, 1, 0, 0, 3)]},
     {"shapes": [(0, 1, 1, 0, 3), (5, 2, 0, 1, 0, 0, 3), (2, 1, 0, 0, 0, 3)]},
 )
-@testing.gpu
 class TestBroadcast(unittest.TestCase):
     def _broadcast(self, xp, dtype, shapes):
         arrays = [testing.shaped_arange(s, xp, dtype) for s in shapes]
@@ -296,9 +301,9 @@ def _broadcast(self, xp, dtype, shapes):
     def test_broadcast(self, dtype):
         broadcast_np = self._broadcast(numpy, dtype, self.shapes)
         broadcast_cp = self._broadcast(cupy, dtype, self.shapes)
-        self.assertEqual(broadcast_np.shape, broadcast_cp.shape)
-        self.assertEqual(broadcast_np.size, broadcast_cp.size)
-        self.assertEqual(broadcast_np.nd, broadcast_cp.nd)
+        assert broadcast_np.shape == broadcast_cp.shape
+        assert broadcast_np.size == broadcast_cp.size
+        assert broadcast_np.nd == broadcast_cp.nd
 
     @testing.for_all_dtypes()
     @testing.numpy_cupy_array_equal()
@@ -329,7 +334,6 @@ def test_broadcast_arrays(self, xp, dtype):
     },
     {"shapes": [(0,), (2,)]},
 )
-@testing.gpu
 class TestInvalidBroadcast(unittest.TestCase):
     @testing.for_all_dtypes()
     def test_invalid_broadcast(self, dtype):