Add attr.gpu to some test functions
@@ -42,6 +42,7 @@ class TestCuda(unittest.TestCase):
x = cuda.cupy.array([]).reshape((0, 10))
self.assertIs(cuda.get_device(x), cuda.DummyDevice)
+ @attr.gpu
def test_get_device_for_int_warning(self):
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter('always')
@@ -62,6 +63,7 @@ class TestCuda(unittest.TestCase):
self.assertIn('get_device is deprecated. Please use get_device_from_id'
' or get_device_from_array instead.', str(w[0].message))
+ @attr.gpu
def test_get_device_for_device_warning(self):
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter('always')
