diff --git a/test/test_nvidia.py b/test/test_nvidia.py index 6fb7e63..b5ae2de 100644 --- a/test/test_nvidia.py +++ b/test/test_nvidia.py @@ -250,6 +250,7 @@ def test_nvidia_env_subs(self): p.get_environment_subs(mock_cliargs) self.assertEqual(cm.exception.code, 1) +@pytest.mark.docker class CudaTest(unittest.TestCase): @classmethod def setUpClass(self): @@ -286,7 +287,6 @@ def setUp(self): em.Interpreter._wasProxyInstalled = False - @pytest.mark.docker def test_no_cuda(self): for tag in self.dockerfile_tags: dig = DockerImageGenerator([], {}, tag) @@ -294,7 +294,6 @@ def test_no_cuda(self): self.assertNotEqual(dig.run(), 0) dig.clear_image() - @pytest.mark.docker def test_cuda_install(self): plugins = list_plugins() desired_plugins = ['cuda']