Skip to content

Commit 8045327

Browse files
author
Diego Ardila
committed
fix tests (new payload field + missing ** -> json)
1 parent d9ce96a commit 8045327

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tests/test_dataset.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,10 +168,12 @@ def test_dataset_append_async(dataset: Dataset):
168168
job = dataset.append(make_dataset_items(), asynchronous=True)
169169
job.sleep_until_complete()
170170
status = job.status()
171+
status["message"]["Payload"] = ""
171172
assert status == {
172173
"job_id": job.id,
173174
"status": "Completed",
174175
"message": {
176+
"Payload": "",
175177
"image_upload_step": {"errored": 0, "pending": 0, "completed": 5},
176178
"started_image_processing": f"Dataset: {dataset.id}, Job: {job.id}",
177179
"ingest_to_reupload_queue": {
@@ -200,10 +202,13 @@ def test_dataset_append_async_with_1_bad_url(dataset: Dataset):
200202
job = dataset.append(ds_items, asynchronous=True)
201203
with pytest.raises(JobError):
202204
job.sleep_until_complete()
203-
assert job.status() == {
205+
status = job.status()
206+
status["message"]["Payload"] = ""
207+
assert status == {
204208
"job_id": f"{job.id}",
205209
"status": "Errored",
206210
"message": {
211+
"Payload": "",
207212
"final_error": (
208213
"One or more of the images you attempted to upload did not process"
209214
" correctly. Please see the status for an overview and the errors for "
@@ -273,7 +278,7 @@ def test_dataset_export_autotag_scores(CLIENT):
273278
def test_annotate_async(dataset: Dataset):
274279
dataset.append(make_dataset_items())
275280
semseg = SegmentationAnnotation.from_json(TEST_SEGMENTATION_ANNOTATIONS[0])
276-
polygon = PolygonAnnotation(**TEST_POLYGON_ANNOTATIONS[0])
281+
polygon = PolygonAnnotation.from_json(TEST_POLYGON_ANNOTATIONS[0])
277282
bbox = BoxAnnotation(**TEST_BOX_ANNOTATIONS[0])
278283

279284
job: AsyncJob = dataset.annotate(
@@ -306,7 +311,7 @@ def test_annotate_async(dataset: Dataset):
306311
def test_annotate_async_with_error(dataset: Dataset):
307312
dataset.append(make_dataset_items())
308313
semseg = SegmentationAnnotation.from_json(TEST_SEGMENTATION_ANNOTATIONS[0])
309-
polygon = PolygonAnnotation(**TEST_POLYGON_ANNOTATIONS[0])
314+
polygon = PolygonAnnotation.from_json(TEST_POLYGON_ANNOTATIONS[0])
310315
bbox = BoxAnnotation(**TEST_BOX_ANNOTATIONS[0])
311316
bbox.reference_id = "fake_garbage"
312317

0 commit comments

Comments
 (0)