tests

Author: ananthsub

tests/accelerators/test_accelerator_connector.py

@@ -345,6 +345,10 @@ class Accel(Accelerator):
         def auto_device_count() -> int:
             return 1
 
+        @staticmethod
+        def is_available() -> bool:
+            return True
+
     class Prec(PrecisionPlugin):
         pass
 

tests/accelerators/test_ddp.py

@@ -91,11 +91,14 @@ def test_torch_distributed_backend_env_variables(tmpdir):
 
 
 @RunIf(skip_windows=True)
-@mock.patch("torch.cuda.device_count", return_value=1)
-@mock.patch("torch.cuda.is_available", return_value=True)
 @mock.patch("torch.cuda.set_device")
+@mock.patch("torch.cuda.is_available", return_value=True)
+@mock.patch("torch.cuda.device_count", return_value=1)
+@mock.patch("pytorch_lightning.accelerators.gpu.GPUAccelerator.is_available", return_value=True)
 @mock.patch.dict(os.environ, {"PL_TORCH_DISTRIBUTED_BACKEND": "gloo"}, clear=True)
-def test_ddp_torch_dist_is_available_in_setup(mock_set_device, mock_is_available, mock_device_count, tmpdir):
+def test_ddp_torch_dist_is_available_in_setup(
+    mock_set_device, mock_cuda_available, mock_device_count, mock_gpu_is_available, tmpdir
+):
     """Test to ensure torch distributed is available within the setup hook using ddp."""
 
     class TestModel(BoringModel):

tests/accelerators/test_gpu.py

@@ -63,17 +63,5 @@ def test_set_cuda_device(set_device_mock, tmpdir):
 
 
 @RunIf(min_gpus=1)
-def test_real_gpu_availability():
+def test_gpu_availability():
     assert GPUAccelerator.is_available()
-
-
-@mock.patch("torch.cuda.is_available", return_value=True)
-@mock.patch("torch.cuda.device_count", return_value=2)
-def test_gpu_available(*_):
-    assert GPUAccelerator.is_available()
-
-
-@mock.patch("torch.cuda.is_available", return_value=False)
-@mock.patch("torch.cuda.device_count", return_value=0)
-def test_gpu_not_available(*_):
-    assert not GPUAccelerator.is_available()