From 8e132eb7867b7954328a980be570059cd00a4270 Mon Sep 17 00:00:00 2001
From: Nick Mowen <nickmowen213@gmail.com>
Date: Tue, 15 Nov 2022 09:05:05 -0700
Subject: [PATCH] Fix test

---
 frigate/test/test_gpu_stats.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/frigate/test/test_gpu_stats.py b/frigate/test/test_gpu_stats.py
index 896907d83..8bff6b40a 100644
--- a/frigate/test/test_gpu_stats.py
+++ b/frigate/test/test_gpu_stats.py
@@ -17,7 +17,7 @@ class TestGpuStats(unittest.TestCase):
         process.stdout = self.amd_results
         sp.return_value = process
         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")
     def test_nvidia_gpu_stats(self, sp):
@@ -28,8 +28,8 @@ class TestGpuStats(unittest.TestCase):
         nvidia_stats = get_nvidia_gpu_stats()
         assert nvidia_stats == {
             "name": "NVIDIA GeForce RTX 3050",
-            "gpu_usage": "42 %",
-            "memory_usage": "61.5 %",
+            "gpu": "42 %",
+            "mem": "61.5 %",
         }
 
     @patch("subprocess.run")
@@ -40,6 +40,6 @@ class TestGpuStats(unittest.TestCase):
         sp.return_value = process
         intel_stats = get_intel_gpu_stats()
         assert intel_stats == {
-            "gpu_usage": "10.73 %",
-            "memory_usage": "- %",
+            "gpu": "10.73 %",
+            "mem": "- %",
         }