Skip to content

Commit 72b36e1

Browse files
committed
Update test_gpu.py
1 parent 274c9aa commit 72b36e1

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/accelerators/test_gpu.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,17 +63,17 @@ def test_set_cuda_device(set_device_mock, tmpdir):
6363

6464

6565
@RunIf(min_gpus=1)
66-
def test_gpu_availability():
66+
def test_real_gpu_availability():
6767
assert GPUAccelerator.is_available()
6868

6969

7070
@mock.patch("torch.cuda.is_available", return_value=True)
7171
@mock.patch("torch.cuda.device_count", return_value=2)
72-
def test_mocked_gpu_available(*_):
72+
def test_gpu_available(*_):
7373
assert GPUAccelerator.is_available()
7474

7575

7676
@mock.patch("torch.cuda.is_available", return_value=False)
7777
@mock.patch("torch.cuda.device_count", return_value=0)
78-
def test_mocked_gpu_availability(*_):
78+
def test_gpu_not_available(*_):
7979
assert not GPUAccelerator.is_available()

0 commit comments

Comments
 (0)