CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-06 18:47:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

add memory metrics

Browse Source
This commit is contained in:
harisreedhar
2026-02-10 17:05:56 +05:30
committed by henryruhs
parent 2624c281eb
commit 14b1ed6b76
3 changed files with 62 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_metrics.py
+26
View File
@@ -42,6 +42,24 @@ def mock_detect_execution_devices(mocker : MockerFixture) -> None:
}
}
])
mocker.patch('facefusion.system.detect_memory_metrics', return_value =
{
'total':
{
'value': 32,
'unit': 'GB'
},
'free':
{
'value': 16,
'unit': 'GB'
},
'utilization':
{
'value': 50,
'unit': '%'
}
})
mocker.patch('facefusion.system.detect_execution_devices', return_value =
[
{
@@ -126,6 +144,10 @@ def test_get_metrics(test_client : TestClient) -> None:
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('memory').get('total').get('value') == 32
assert metrics_body.get('memory').get('free').get('unit') == 'GiB'
assert metrics_body.get('memory').get('utilization').get('value') == 50
def test_websocket_metrics(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
@@ -147,3 +169,7 @@ def test_websocket_metrics(test_client : TestClient) -> None:
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('utilization').get('value') == 60
assert metrics_set.get('memory').get('total').get('value') == 32
assert metrics_set.get('memory').get('free').get('unit') == 'GiB'
assert metrics_set.get('memory').get('utilization').get('value') == 50