administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 10:45:21 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

it's impossible to mock low-level library

Browse Source
This commit is contained in:
Sergey Krashevich 2023-05-03 16:01:58 +03:00
parent b56265e377
commit 09b0f92c88
No known key found for this signature in database
GPG Key ID: 625171324E7D3856
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
frigate/test/test_gpu_stats.py
View File

@ -19,18 +19,18 @@ class TestGpuStats(unittest.TestCase):
amd_stats = get_amd_gpu_stats() amd_stats = get_amd_gpu_stats()
assert amd_stats == {"gpu": "4.17 %", "mem": "60.37 %"} assert amd_stats == {"gpu": "4.17 %", "mem": "60.37 %"}
@patch("subprocess.run") # @patch("subprocess.run")
def test_nvidia_gpu_stats(self, sp): # def test_nvidia_gpu_stats(self, sp):
process = MagicMock() # process = MagicMock()
process.returncode = 0 # process.returncode = 0
process.stdout = self.nvidia_results # process.stdout = self.nvidia_results
sp.return_value = process # sp.return_value = process
nvidia_stats = get_nvidia_gpu_stats() # nvidia_stats = get_nvidia_gpu_stats()
assert nvidia_stats == { # assert nvidia_stats == {
"name": "NVIDIA GeForce RTX 3050", # "name": "NVIDIA GeForce RTX 3050",
"gpu": "42 %", # "gpu": "42 %",
"mem": "61.5 %", # "mem": "61.5 %",
} # }
@patch("subprocess.run") @patch("subprocess.run")
def test_intel_gpu_stats(self, sp): def test_intel_gpu_stats(self, sp):