Skip to content

Commit a1bdde0

Browse files
author
Matt Sokoloff
committed
bug fix
1 parent d81e99e commit a1bdde0

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

tests/integration/test_data_row_metadata.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,7 @@ def mdo(client):
2121

2222

2323
@pytest.fixture
24-
def big_dataset(client, dataset: Dataset, image_url):
25-
image_url = client
24+
def big_dataset(dataset: Dataset, image_url):
2625
task = dataset.create_data_rows([
2726
{
2827
"row_data": image_url,
@@ -97,13 +96,11 @@ def test_bulk_delete_datarow_metadata(datarow, mdo):
9796
mdo.bulk_upsert([metadata])
9897

9998
assert len(datarow.metadata["fields"])
100-
101-
mdo.bulk_delete([
102-
DeleteDataRowMetadata(data_row_id=datarow.uid,
103-
fields=[m.schema_id for m in metadata.fields])
104-
])
105-
106-
assert not len(datarow.metadata["fields"])
99+
upload_ids = [m.schema_id for m in metadata.fields]
100+
mdo.bulk_delete(
101+
[DeleteDataRowMetadata(data_row_id=datarow.uid, fields=upload_ids)])
102+
remaining_ids = set([f['schema_id'] for f in datarow.metadata["fields"]])
103+
assert not len(remaining_ids.intersection(set(upload_ids)))
107104

108105

109106
def test_bulk_partial_delete_datarow_metadata(datarow, mdo):

0 commit comments

Comments
 (0)