Skip to content

Commit bf0de29

Browse files
Don't iterate over all instances in the project in test_iterate_over_instances (#1383)
1 parent 718014f commit bf0de29

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tests/tests_integration/test_api/test_data_modeling/test_instances.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -304,9 +304,14 @@ def test_retrieve_non_existent(self, cognite_client: CogniteClient) -> None:
304304
assert cognite_client.data_modeling.instances.retrieve(("myNonExistingSpace", "myImaginaryNode")).nodes == []
305305

306306
def test_iterate_over_instances(self, cognite_client: CogniteClient) -> None:
307-
for nodes in cognite_client.data_modeling.instances(chunk_size=2, limit=-1):
308-
assert isinstance(nodes, NodeList)
309-
assert len(nodes) <= 2
307+
iterator = cognite_client.data_modeling.instances(chunk_size=2)
308+
first_iter = next(iterator)
309+
assert isinstance(first_iter, NodeList)
310+
assert len(first_iter) <= 2
311+
312+
second_iter = next(iterator)
313+
assert isinstance(second_iter, NodeList)
314+
assert len(second_iter) <= 2
310315

311316
def test_apply_invalid_node_data(self, cognite_client: CogniteClient, person_view: View) -> None:
312317
# Arrange

0 commit comments

Comments
 (0)