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

Fix test

Browse Source
This commit is contained in:
Nick Mowen 2022-11-15 09:05:05 -07:00
parent ea63fcd431
commit 8e132eb786
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/test/test_gpu_stats.py
View File

@ -17,7 +17,7 @@ class TestGpuStats(unittest.TestCase):
process.stdout = self.amd_results process.stdout = self.amd_results
sp.return_value = process sp.return_value = process
amd_stats = get_amd_gpu_stats() amd_stats = get_amd_gpu_stats()
assert amd_stats == {"gpu_usage": "4.17 %", "memory_usage": "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):
@ -28,8 +28,8 @@ class TestGpuStats(unittest.TestCase):
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_usage": "42 %", "gpu": "42 %",
"memory_usage": "61.5 %", "mem": "61.5 %",
} }
@patch("subprocess.run") @patch("subprocess.run")
@ -40,6 +40,6 @@ class TestGpuStats(unittest.TestCase):
sp.return_value = process sp.return_value = process
intel_stats = get_intel_gpu_stats() intel_stats = get_intel_gpu_stats()
assert intel_stats == { assert intel_stats == {
"gpu_usage": "10.73 %", "gpu": "10.73 %",
"memory_usage": "- %", "mem": "- %",
} }