From 84ac47a8ab196ee20ad778219bfa13df48f87241 Mon Sep 17 00:00:00 2001 From: "Danilo M." Date: Mon, 9 Nov 2026 18:59:58 +0100 Subject: feat: Intel xe GPU stats (temp, fan, VRAM) --- tests/test_stats.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'tests') diff --git a/tests/test_stats.py b/tests/test_stats.py index fad6cc2..81acdba 100644 --- a/tests/test_stats.py +++ b/tests/test_stats.py @@ -21,5 +21,28 @@ class Stats(unittest.TestCase): self.assertIn("stats", names) +class Intel(unittest.TestCase): + def test_parse_bytes(self): + self.assertEqual(stats._parse_bytes("12G"), 12 * 1024 ** 3) + self.assertEqual(stats._parse_bytes("512M"), 512 * 1024 ** 2) + self.assertEqual(stats._parse_bytes(2048), 2048) + self.assertIsNone(stats._parse_bytes("nope")) + + def test_vram_scan_counts_each_client_once(self): + import tempfile + root = Path(tempfile.mkdtemp()) + rows = [ + ("100", "3", "drm-driver:\txe\ndrm-client-id:\t7\ndrm-resident-vram0:\t1024 MiB\n"), + ("100", "4", "drm-driver:\txe\ndrm-client-id:\t7\ndrm-resident-vram0:\t1024 MiB\n"), + ("200", "5", "drm-driver:\txe\ndrm-client-id:\t8\ndrm-resident-vram0:\t512 MiB\n"), + ("300", "6", "drm-driver:\ti915\ndrm-client-id:\t9\ndrm-resident-vram0:\t999 MiB\n"), + ] + for pid, fd, body in rows: + d = root / pid / "fdinfo" + d.mkdir(parents=True, exist_ok=True) + (d / fd).write_text(body) + self.assertEqual(stats._intel_vram_used(root), (1024 + 512) * 1024 ** 2) + + if __name__ == "__main__": unittest.main() -- cgit v1.2.3