Refactor get datarows tests so that I can reuse some fixtures

Author: Val Brodsky

libs/labelbox/tests/integration/conftest.py

@@ -7,6 +7,15 @@
 from typing import Type
 
 import pytest
+<<<<<<< HEAD
+=======
+from constants import (
+    CAPTURE_DT_SCHEMA_ID,
+    SPLIT_SCHEMA_ID,
+    TEST_SPLIT_ID,
+    TEXT_SCHEMA_ID,
+)
+>>>>>>> a4fe5b93 (Refactor get datarows tests so that I can reuse some fixtures)
 
 from labelbox import (
     Classification,
@@ -24,6 +33,7 @@
 from labelbox.schema.ontology_kind import OntologyKind
 from labelbox.schema.queue_mode import QueueMode
 from labelbox.schema.user import User
+<<<<<<< HEAD
 
 
 @pytest.fixture
@@ -48,6 +58,8 @@ def constants():
         "EXPECTED_METADATA_SCHEMA_IDS": EXPECTED_METADATA_SCHEMA_IDS,
         "CUSTOM_TEXT_SCHEMA_NAME": CUSTOM_TEXT_SCHEMA_NAME,
     }
+=======
+>>>>>>> a4fe5b93 (Refactor get datarows tests so that I can reuse some fixtures)
 
 
 @pytest.fixture

libs/labelbox/tests/integration/test_data_rows.py

@@ -405,7 +405,7 @@ def test_create_data_row_with_metadata_dict(
         row_data=image_url, metadata_fields=make_metadata_fields_dict
     )
 
-    assert len(list(dataset.data_rows())) == 1
+    assert len([dr for dr in dataset.data_rows()]) == 1
     assert data_row.dataset() == dataset
     assert data_row.created_by() == client.get_user()
     assert data_row.organization() == client.get_organization()