Align compact GPU metrics

This commit is contained in:
2026-06-25 11:25:42 +08:00
parent 77c22ab21f
commit b8f007fd5c
2 changed files with 29 additions and 13 deletions

View File

@@ -2,7 +2,14 @@ from pathlib import Path
from tempfile import TemporaryDirectory
import unittest
from mm.cli import load_machines, parse_nvidia_smi_csv
from mm.cli import (
GPUStat,
MachineConfig,
MachineResult,
gpu_summary,
load_machines,
parse_nvidia_smi_csv,
)
class ConfigTests(unittest.TestCase):
@@ -59,5 +66,21 @@ class NvidiaSmiParsingTests(unittest.TestCase):
self.assertEqual(gpus[1].power_draw_w, 49.8)
class RenderingTests(unittest.TestCase):
def test_gpu_summary_uses_fixed_width_decimal_metrics(self) -> None:
result = MachineResult(
machine=MachineConfig(alias="dash0", label="dash0"),
gpus=(
GPUStat("0", "GPU", 88 * 1024, 96 * 1024, 0, None, None, None),
GPUStat("1", "GPU", 0, 96 * 1024, 0, None, None, None),
),
)
self.assertEqual(
gpu_summary(result).plain,
"0 88.0/ 96.0G 0.0% 1 0.0/ 96.0G 0.0%",
)
if __name__ == "__main__":
unittest.main()