diff --git a/tests/integration/test_data_rows.py b/tests/integration/test_data_rows.py index eb7895238..0b4563f0d 100644 --- a/tests/integration/test_data_rows.py +++ b/tests/integration/test_data_rows.py @@ -149,3 +149,16 @@ def test_data_row_deletion(dataset, rand_gen): data_rows = list(dataset.data_rows()) assert {dr.external_id for dr in data_rows} == expected + + +def test_data_row_iteration(dataset, rand_gen) -> None: + task = dataset.create_data_rows([ + { + DataRow.row_data: IMG_URL + }, + { + "row_data": IMG_URL + }, + ]) + task.wait_till_done() + data_row = next(dataset.data_rows())