Skip to content

Commit db81ff8

Browse files
Merge pull request #812 from Labelbox/kkim/better-error-handling-label-creation
[AL-0] Better error handling for label creation during test
2 parents 496e7a2 + 0f31c62 commit db81ff8

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

tests/integration/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -339,17 +339,17 @@ def configured_project_with_label(client, rand_gen, image_url, project, dataset,
339339

340340
def create_label():
341341
""" Ad-hoc function to create a LabelImport
342-
343342
Creates a LabelImport task which will create a label
344343
"""
345344
upload_task = LabelImport.create_from_objects(
346345
client, project.uid, f'label-import-{uuid.uuid4()}', predictions)
347346
upload_task.wait_until_done(sleep_time_seconds=5)
348-
assert upload_task.state == AnnotationImportState.FINISHED
347+
assert upload_task.state == AnnotationImportState.FINISHED, "Label Import failed"
349348

350349
project.create_label = create_label
351350
project.create_label()
352-
label = next(project.labels())
351+
label = project.labels().get_one()
352+
assert label is not None, "Cannot fetch created label"
353353
yield [project, dataset, datarow, label]
354354

355355
for label in project.labels():

tests/integration/test_data_row_media_attributes.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@ def test_export_empty_media_attributes(configured_project_with_label):
55
project, _, _, _ = configured_project_with_label
66
# Wait for exporter to retrieve latest labels
77
sleep(10)
8-
labels = project.label_generator()
9-
label = next(labels)
10-
assert label.data.media_attributes == {}
8+
labels = list(project.label_generator())
9+
assert len(
10+
labels
11+
) == 1, "Label export job unexpectedly returned an empty result set`"
12+
assert labels[0].data.media_attributes == {}

0 commit comments

Comments
 (0)