Skip to content

Commit a8beb0c

Browse files
committed
Fix mock test for nvml after moving NVML module
1 parent 4801627 commit a8beb0c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/test_nvml_mocked.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def setup_mock(nvml):
1818
return nvml
1919

2020

21-
@patch('kernel_tuner.nvml.nvml')
21+
@patch('kernel_tuner.observers.nvml.nvml')
2222
def test_get_nvml_pwr_limits(nvml):
2323
nvml = setup_mock(nvml)
2424
result = get_nvml_pwr_limits(0, quiet=True)
@@ -30,7 +30,7 @@ def test_get_nvml_pwr_limits(nvml):
3030
assert result['nvml_pwr_limit'][-1] == 150
3131

3232

33-
@patch('kernel_tuner.nvml.nvml')
33+
@patch('kernel_tuner.observers.nvml.nvml')
3434
def test_get_nvml_gr_clocks(nvml):
3535
nvml = setup_mock(nvml)
3636
result = get_nvml_gr_clocks(0, quiet=True)
@@ -40,15 +40,15 @@ def test_get_nvml_gr_clocks(nvml):
4040
assert result['nvml_gr_clock'] == [1000, 3000]
4141

4242

43-
@patch('kernel_tuner.nvml.nvml')
43+
@patch('kernel_tuner.observers.nvml.nvml')
4444
def test_get_nvml_mem_clocks(nvml):
4545
nvml = setup_mock(nvml)
4646
result = get_nvml_mem_clocks(0, quiet=False)
4747
print(result)
4848
assert result['nvml_mem_clock'] == [2100]
4949

5050

51-
@patch('kernel_tuner.nvml.nvml')
51+
@patch('kernel_tuner.observers.nvml.nvml')
5252
def test_get_idle_power(nvml):
5353
nvml = setup_mock(nvml)
5454
result = get_idle_power(0)

0 commit comments

Comments
 (0)