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

clean and simplify tests

Browse Source
This commit is contained in:
henryruhs
2026-05-29 15:44:07 +02:00
parent 2a8672b54d
commit 871559cb6a
3 changed files with 10 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_stream.py
+1 -8
View File
@@ -119,6 +119,7 @@ def test_stream_video(test_client : TestClient, video_codec : VideoCodec) -> Non
if video_codec == 'av1':
rtc.add_video_track(peer_connection, 'sendrecv', video_codec, 35)
if video_codec == 'vp8':
rtc.add_video_track(peer_connection, 'sendrecv', video_codec, 96)
@@ -136,14 +137,6 @@ def test_stream_video(test_client : TestClient, video_codec : VideoCodec) -> Non
assert stream_response.status_code == 201
assert 'm=video' in stream_response.text
session_id = session_manager.find_session_id(access_token)
for peer in rtc_store.get_peers(session_id):
bitrate = peer.get('bitrate')
assert bitrate.value == 0
rtc.handle_remb(0, 6000000, ctypes.addressof(bitrate))
assert bitrate.value == 6000
def test_delete_stream_video(test_client : TestClient) -> None:
create_session_response = test_client.post('/session', json =
tests/test_rtc.py
+3
View File
@@ -154,6 +154,9 @@ def test_wire_remb(video_codec : VideoCodec, payload_type : int) -> None:
}
wire_remb(video_sender_track, rtc_peer.get('bitrate'))
assert rtc_peer.get('bitrate').value == 0
handle_remb(0, 6000000, ctypes.addressof(rtc_peer.get('bitrate')))
assert rtc_peer.get('bitrate').value == 6000
tests/test_stream_helper.py
+6 -7
View File
@@ -101,22 +101,21 @@ def test_create_and_destroy_video_encoder(video_codec : VideoCodec) -> None:
resolution = (vision_frame.shape[1], vision_frame.shape[0])
frame_buffer = cv2.cvtColor(vision_frame, cv2.COLOR_BGR2YUV_I420).tobytes()
encoder = create_video_encoder(video_codec, resolution, 4000)
assert encoder is not None
video_encoder = create_video_encoder(video_codec, resolution, 4000)
if video_codec == 'av1':
assert aom_encoder.encode(encoder, frame_buffer, resolution, 0)
assert aom_encoder.encode(video_encoder, frame_buffer, resolution, 0)
if video_codec == 'vp8':
assert vpx_encoder.encode(encoder, frame_buffer, resolution, 0)
assert vpx_encoder.encode(video_encoder, frame_buffer, resolution, 0)
destroy_video_encoder(video_codec, encoder)
destroy_video_encoder(video_codec, video_encoder)
if video_codec == 'av1':
assert not aom_encoder.encode(encoder, frame_buffer, resolution, 1)
assert not aom_encoder.encode(video_encoder, frame_buffer, resolution, 1)
if video_codec == 'vp8':
assert not vpx_encoder.encode(encoder, frame_buffer, resolution, 1)
assert not vpx_encoder.encode(video_encoder, frame_buffer, resolution, 1)
# TODO: refine test