Fix skipUnless
@@ -44,10 +44,10 @@ class TestCuda(unittest.TestCase):
self.assertIs(cuda.get_device(x), cuda.DummyDevice)
@attr.gpu
- @unittest.skipUnless(six.PY3)
+ @unittest.skipUnless(
+ six.PY3, 'Python2.7 has a bug in catch_warnings, so this test is '
+ 'skipped for Python2.7')
def test_get_device_warning(self):
- # NOTE: Python2.7 has a bug in catch_warnings, so this test is skipped
- # for Python2.7
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter('always')
cuda.get_device(cuda.cupy.array([1]))
