Skip to content

Commit 1f7f643

Browse files
committed
Update Unit tests
1 parent 2b8a5c6 commit 1f7f643

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

tests/integration/test_data_rows.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def test_lookup_data_rows(client, dataset):
133133
assert all([len(x) == 1 for x in lookup.values()])
134134
assert lookup[uid][0] == dr.uid
135135
assert lookup[uid2][0] == dr2.uid
136-
#1 external id : 2 uid
136+
# 1 external id : 2 uid
137137
dr3 = dataset.create_data_row(row_data="123", external_id=uid2)
138138
lookup = client.get_data_row_ids_for_external_ids([uid2])
139139
assert len(lookup) == 1
@@ -307,7 +307,7 @@ def test_create_data_row_with_metadata(mdo, dataset, image_url):
307307
assert len(metadata_fields) == 3
308308
assert len(metadata) == 3
309309
assert [m["schemaId"] for m in metadata_fields
310-
].sort() == EXPECTED_METADATA_SCHEMA_IDS
310+
].sort() == EXPECTED_METADATA_SCHEMA_IDS
311311
for m in metadata:
312312
assert mdo._parse_upsert(m)
313313

@@ -331,7 +331,7 @@ def test_create_data_row_with_metadata_dict(mdo, dataset, image_url):
331331
assert len(metadata_fields) == 3
332332
assert len(metadata) == 3
333333
assert [m["schemaId"] for m in metadata_fields
334-
].sort() == EXPECTED_METADATA_SCHEMA_IDS
334+
].sort() == EXPECTED_METADATA_SCHEMA_IDS
335335
for m in metadata:
336336
assert mdo._parse_upsert(m)
337337

@@ -388,7 +388,7 @@ def test_create_data_rows_with_metadata(mdo, dataset, image_url):
388388
assert len(metadata_fields) == 3
389389
assert len(metadata) == 3
390390
assert [m["schemaId"] for m in metadata_fields
391-
].sort() == EXPECTED_METADATA_SCHEMA_IDS
391+
].sort() == EXPECTED_METADATA_SCHEMA_IDS
392392
for m in metadata:
393393
assert mdo._parse_upsert(m)
394394

@@ -623,7 +623,8 @@ def test_data_row_iteration(dataset, image_url) -> None:
623623
def test_data_row_attachments(dataset, image_url):
624624
attachments = [("IMAGE", image_url, "attachment image"),
625625
("TEXT", "test-text", None),
626-
("IMAGE_OVERLAY", image_url, "Overlay"), ("HTML", image_url)]
626+
("IMAGE_OVERLAY", image_url, "Overlay"),
627+
("HTML", image_url, None)]
627628
task = dataset.create_data_rows([{
628629
"row_data":
629630
image_url,
@@ -776,7 +777,7 @@ def test_data_row_bulk_creation_with_unique_global_keys(dataset, sample_image):
776777

777778
task.wait_till_done()
778779
assert {row.global_key for row in dataset.data_rows()
779-
} == {global_key_1, global_key_2, global_key_3}
780+
} == {global_key_1, global_key_2, global_key_3}
780781

781782

782783
def test_data_row_bulk_creation_with_same_global_keys(dataset, sample_image):
@@ -864,7 +865,7 @@ def test_data_row_bulk_creation_sync_with_unique_global_keys(
864865
])
865866

866867
assert {row.global_key for row in dataset.data_rows()
867-
} == {global_key_1, global_key_2, global_key_3}
868+
} == {global_key_1, global_key_2, global_key_3}
868869

869870

870871
def test_data_row_rulk_creation_sync_with_same_global_keys(
@@ -933,7 +934,8 @@ def test_create_tiled_layer(dataset, tile_content):
933934
**tile_content, 'media_type': 'TMS_SIMPLE'
934935
},
935936
tile_content,
936-
tile_content['row_data'] # Old way to check for backwards compatibility
937+
# Old way to check for backwards compatibility
938+
tile_content['row_data']
937939
]
938940
dataset.create_data_rows_sync(examples)
939941
data_rows = list(dataset.data_rows())

0 commit comments

Comments
 (0)