some cosmetics

This commit is contained in:
harisreedhar
2026-03-17 14:01:54 +01:00
committed by henryruhs
parent 671eb7129d
commit e4bd360577
+3
View File
@@ -117,9 +117,11 @@ def test_get_metrics(test_client : TestClient) -> None:
metrics_body = metrics_response.json() metrics_body = metrics_response.json()
assert metrics_response.status_code == 200 assert metrics_response.status_code == 200
assert metrics_body.get('execution_devices')[0].get('driver_version') == '555.42' assert metrics_body.get('execution_devices')[0].get('driver_version') == '555.42'
assert metrics_body.get('execution_devices')[0].get('product').get('name') == 'RTX 4090' assert metrics_body.get('execution_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_body.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24 assert metrics_body.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_body.get('disks')[0].get('total').get('value') == 500 assert metrics_body.get('disks')[0].get('total').get('value') == 500
assert metrics_body.get('disks')[0].get('free').get('unit') == 'GB' assert metrics_body.get('disks')[0].get('free').get('unit') == 'GB'
assert metrics_body.get('disks')[0].get('utilization').get('value') == 60 assert metrics_body.get('disks')[0].get('utilization').get('value') == 60
@@ -141,6 +143,7 @@ def test_websocket_metrics(test_client : TestClient) -> None:
assert metrics_set.get('execution_devices')[0].get('driver_version') == '555.42' assert metrics_set.get('execution_devices')[0].get('driver_version') == '555.42'
assert metrics_set.get('execution_devices')[0].get('product').get('name') == 'RTX 4090' assert metrics_set.get('execution_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_set.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24 assert metrics_set.get('execution_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_set.get('disks')[0].get('total').get('value') == 500 assert metrics_set.get('disks')[0].get('total').get('value') == 500
assert metrics_set.get('disks')[0].get('free').get('unit') == 'GB' assert metrics_set.get('disks')[0].get('free').get('unit') == 'GB'
assert metrics_set.get('disks')[0].get('utilization').get('value') == 60 assert metrics_set.get('disks')[0].get('utilization').get('value') == 60