19 lines
507 B
Python
19 lines
507 B
Python
from app.modules.system.service import parse_nvidia_smi_csv
|
|
|
|
|
|
def test_parse_nvidia_smi_csv():
|
|
output = "0, NVIDIA GeForce RTX 4090, 24564, 15, 24069, 0, 37\n"
|
|
gpus = parse_nvidia_smi_csv(output)
|
|
assert gpus == [
|
|
{
|
|
"index": 0,
|
|
"name": "NVIDIA GeForce RTX 4090",
|
|
"memory_total_mb": 24564,
|
|
"memory_used_mb": 15,
|
|
"memory_free_mb": 24069,
|
|
"utilization_gpu_percent": 0,
|
|
"temperature_c": 37,
|
|
}
|
|
]
|
|
|