Insert N docs, iterate, verify count + PK + scalar fields.
(iter_collection)
| 78 | |
| 79 | |
| 80 | def test_iter_docs_basic(iter_collection): |
| 81 | """Insert N docs, iterate, verify count + PK + scalar fields.""" |
| 82 | n = 50 |
| 83 | result = iter_collection.insert(_make_docs(n)) |
| 84 | assert bool(result) |
| 85 | iter_collection.flush() |
| 86 | |
| 87 | seen_ids = set() |
| 88 | count = 0 |
| 89 | for doc in iter_collection.iter_docs(): |
| 90 | assert isinstance(doc, Doc) |
| 91 | assert doc.id != "" |
| 92 | # scalar fields present |
| 93 | assert doc.field("id") is not None |
| 94 | assert doc.field("name") is not None |
| 95 | seen_ids.add(doc.id) |
| 96 | count += 1 |
| 97 | |
| 98 | assert count == n |
| 99 | assert len(seen_ids) == n |
| 100 | assert seen_ids == {f"{i}" for i in range(n)} |
| 101 | |
| 102 | |
| 103 | def test_iter_docs_include_vector(iter_collection): |