|
1 | 1 | import time |
| 2 | + |
| 3 | +import pytest |
2 | 4 | import labelbox as lb |
3 | 5 | from labelbox.data.annotation_types.data.video import VideoData |
4 | 6 | import labelbox.types as lb_types |
5 | 7 | from labelbox.schema.annotation_import import AnnotationImportState |
6 | 8 |
|
7 | 9 |
|
| 10 | +@pytest.fixture |
| 11 | +def user_id(client): |
| 12 | + return client.get_user().uid |
| 13 | + |
| 14 | + |
| 15 | +@pytest.fixture |
| 16 | +def org_id(client): |
| 17 | + return client.get_organization().uid |
| 18 | + |
| 19 | + |
8 | 20 | def test_export_v2_video(client, configured_project_without_data_rows, |
9 | | - video_data, video_data_row, |
10 | | - bbox_video_annotation_objects, rand_gen): |
| 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 |
11 | 27 |
|
12 | 28 | project = configured_project_without_data_rows |
13 | 29 | project_id = project.uid |
14 | 30 | labels = [] |
| 31 | + |
15 | 32 | _, data_row_uids = video_data |
16 | 33 | project.create_batch( |
17 | 34 | rand_gen(str), |
@@ -156,6 +173,7 @@ def test_export_v2_video(client, configured_project_without_data_rows, |
156 | 173 | ) #note we create 1 label per data row, 1 data row so 1 label |
157 | 174 | export_label = project_export_labels[0] |
158 | 175 | assert (export_label['label_kind']) == 'Video' |
| 176 | + |
159 | 177 | assert (export_label['label_details'].keys() |
160 | 178 | ) == expected_export_label['label_details'].keys() |
161 | 179 |
|
|
0 commit comments