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

Add tests and fix AMD formatting

Browse Source
This commit is contained in:
Nick Mowen 2022-09-24 20:04:40 -06:00
parent b0a282236f
commit 9b5516caa5
2 changed files with 25 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
frigate/test/test_gpu_stats.py Normal file
View File

@ -0,0 +1,21 @@
import unittest
from unittest.mock import MagicMock, patch
from frigate.util import get_amd_gpu_stats
class TestGpuStats(unittest.TestCase):
def setUp(self):
self.nvidia_results = "name, utilization.gpu [%], utilization.memory [%]\nNVIDIA GeForce RTX 3050, 41 %, 1 %\n"
self.amd_results = "Unknown Radeon card. <= R500 won't work, new cards might.\nDumping to -, line limit 1.\n1664070990.607556: bus 10, gpu 4.17%, ee 0.00%, vgt 0.00%, ta 0.00%, tc 0.00%, sx 0.00%, sh 0.00%, spi 0.83%, smx 0.00%, cr 0.00%, sc 0.00%, pa 0.00%, db 0.00%, cb 0.00%, vram 60.37% 294.04mb, gtt 0.33% 52.21mb, mclk 100.00% 1.800ghz, sclk 26.65% 0.533ghz\n"
@patch("subprocess.run")
def test_amd_gpu_stats(self, sp):
process = MagicMock()
process.returncode = 0
process.stdout = self.amd_results
sp.return_value = process
amd_stats = get_amd_gpu_stats()
print(amd_stats)
pass

8
frigate/util.py
View File

@ -800,9 +800,9 @@ def get_amd_gpu_stats() -> dict[str, str]:
for hw in usages:
if "gpu" in hw:
results["gpu_usage"] = f"{hw.strip().split(' ')[1].split(' ')[0]} %"
results["gpu_usage"] = f"{hw.strip().split(' ')[1].replace('%', '')} %"
elif "vram" in hw:
results["memory_usage"] = f"{hw.strip().split(' ')[1].split(' ')[0]} %"
results["memory_usage"] = f"{hw.strip().split(' ')[1].replace('%', '')} %"
return results
@ -826,9 +826,9 @@ def get_intel_gpu_stats() -> dict[str, str]:
for hw in usages:
if "gpu" in hw:
results["gpu_usage"] = f"{hw.strip().split(' ')[1].replace('%', '')} %"
results["gpu_usage"] = f"{hw.strip().split(' ')[1].split(' ')[0]} %"
elif "vram" in hw:
results["memory_usage"] = f"{hw.strip().split(' ')[1].replace('%', '')} %"
results["memory_usage"] = f"{hw.strip().split(' ')[1].split(' ')[0]} %"
return results