From bf0de29e48e22252f8661f99475525a8904d2eb5 Mon Sep 17 00:00:00 2001 From: Erlend vollset Date: Thu, 28 Sep 2023 11:48:28 +0200 Subject: [PATCH] Don't iterate over all instances in the project in test_iterate_over_instances (#1383) --- .../test_api/test_data_modeling/test_instances.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/tests/tests_integration/test_api/test_data_modeling/test_instances.py b/tests/tests_integration/test_api/test_data_modeling/test_instances.py index 4dcbf8cd45..8498b9e826 100644 --- a/tests/tests_integration/test_api/test_data_modeling/test_instances.py +++ b/tests/tests_integration/test_api/test_data_modeling/test_instances.py @@ -304,9 +304,14 @@ def test_retrieve_non_existent(self, cognite_client: CogniteClient) -> None: assert cognite_client.data_modeling.instances.retrieve(("myNonExistingSpace", "myImaginaryNode")).nodes == [] def test_iterate_over_instances(self, cognite_client: CogniteClient) -> None: - for nodes in cognite_client.data_modeling.instances(chunk_size=2, limit=-1): - assert isinstance(nodes, NodeList) - assert len(nodes) <= 2 + iterator = cognite_client.data_modeling.instances(chunk_size=2) + first_iter = next(iterator) + assert isinstance(first_iter, NodeList) + assert len(first_iter) <= 2 + + second_iter = next(iterator) + assert isinstance(second_iter, NodeList) + assert len(second_iter) <= 2 def test_apply_invalid_node_data(self, cognite_client: CogniteClient, person_view: View) -> None: # Arrange