From b56b58f5125585dbf1bfc9fdc31d889220f24ed1 Mon Sep 17 00:00:00 2001
From: rllin <randall@labelbox.com>
Date: Wed, 21 Oct 2020 11:19:54 -0700
Subject: [PATCH] test coverage

---
 tests/integration/test_data_rows.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

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())