Skip to content

Commit 4f33e96

Browse files
committed
Add export dataset tests
1 parent 784a121 commit 4f33e96

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/integration/test_dataset.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,20 @@ def test_data_row_export(dataset, image_url):
148148
assert set(result) == ids
149149

150150

151+
def test_dataset_export_v2(dataset, image_url):
152+
n_data_rows = 5
153+
ids = set()
154+
for _ in range(n_data_rows):
155+
ids.add(dataset.create_data_row(row_data=image_url))
156+
task = dataset.export_v2(params={
157+
"performance_details": False,
158+
"label_details": True
159+
})
160+
assert task.status == "COMPLETE"
161+
assert task.errors is None
162+
assert len(task.result) == n_data_rows
163+
164+
151165
def test_create_descriptor_file(dataset):
152166
import unittest.mock as mock
153167
with mock.patch.object(dataset.client,

0 commit comments

Comments
 (0)