Skip to content

Commit 63c54f5

Browse files
authored
Update the exporter video test to check for frame rate and count (#1254)
2 parents c01a9c0 + ef0190c commit 63c54f5

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

tests/integration/export_v2/test_export_video.py

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,14 @@ def org_id(client):
1717
return client.get_organization().uid
1818

1919

20-
def test_export_v2_video(client, configured_project_without_data_rows,
21-
video_data, video_data_row, is_adv_enabled,
22-
bbox_video_annotation_objects, rand_gen, user_id,
23-
org_id):
24-
25-
orgid = client.get_organization().uid
26-
userid = client.get_user().uid
20+
def test_export_v2_video(
21+
client,
22+
configured_project_without_data_rows,
23+
video_data,
24+
video_data_row,
25+
bbox_video_annotation_objects,
26+
rand_gen,
27+
):
2728

2829
project = configured_project_without_data_rows
2930
project_id = project.uid
@@ -51,6 +52,7 @@ def test_export_v2_video(client, configured_project_without_data_rows,
5152

5253
num_retries = 5
5354
task = None
55+
5456
while (num_retries > 0):
5557
task = project.export_v2(
5658
params={
@@ -72,7 +74,10 @@ def test_export_v2_video(client, configured_project_without_data_rows,
7274
assert data_row_export['global_key'] == video_data_row['global_key']
7375
assert data_row_export['row_data'] == video_data_row['row_data']
7476
assert export_data[0]['media_attributes']['mime_type'] == 'video/mp4'
75-
77+
assert export_data[0]['media_attributes'][
78+
'frame_rate'] == 10 # as per the video_data fixture
79+
assert export_data[0]['media_attributes'][
80+
'frame_count'] == 100 # as per the video_data fixture
7681
expected_export_label = {
7782
'label_kind': 'Video',
7883
'version': '1.0.0',

0 commit comments

Comments
 (0)