From 7ab6b539b381d00d7f3f242a2364c54411fa93a2 Mon Sep 17 00:00:00 2001 From: Attila Papai Date: Thu, 3 Aug 2023 16:29:11 +0200 Subject: [PATCH 1/2] fix assert message --- tests/integration/test_data_rows.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/integration/test_data_rows.py b/tests/integration/test_data_rows.py index a70c1c41b..e9f18035d 100644 --- a/tests/integration/test_data_rows.py +++ b/tests/integration/test_data_rows.py @@ -998,7 +998,8 @@ def test_create_data_row_with_media_type(dataset, image_url): with pytest.raises(labelbox.exceptions.InvalidQueryError) as exc: dr = dataset.create_data_row( row_data={'invalid_object': 'invalid_value'}, media_type="IMAGE") - assert "Found invalid contents for media type: \'IMAGE\'" in str(exc.value) + assert "Media type validation failed, expected: 'image/*', was: application/json" in str( + exc.value) dataset.create_data_row(row_data=image_url, media_type="IMAGE") From 714dd4c2dddaf2456733a215834c08327cf14dcf Mon Sep 17 00:00:00 2001 From: Attila Papai Date: Thu, 3 Aug 2023 16:30:36 +0200 Subject: [PATCH 2/2] fix assert message --- tests/integration/test_data_rows.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/integration/test_data_rows.py b/tests/integration/test_data_rows.py index e9f18035d..779b39106 100644 --- a/tests/integration/test_data_rows.py +++ b/tests/integration/test_data_rows.py @@ -994,12 +994,16 @@ def test_create_data_row_with_attachments(dataset): assert len(attachments) == 1 -def test_create_data_row_with_media_type(dataset, image_url): +def test_create_data_row_with_media_type(dataset, image_url, is_adv_enabled): with pytest.raises(labelbox.exceptions.InvalidQueryError) as exc: dr = dataset.create_data_row( row_data={'invalid_object': 'invalid_value'}, media_type="IMAGE") - assert "Media type validation failed, expected: 'image/*', was: application/json" in str( - exc.value) + if is_adv_enabled: + assert "Media type validation failed, expected: 'image/*', was: application/json" in str( + exc.value) + else: + assert "Found invalid contents for media type: \'IMAGE\'" in str( + exc.value) dataset.create_data_row(row_data=image_url, media_type="IMAGE")