use iterator instead of list to go over partials

This commit is contained in:
Sarah Hoffmann
2025-04-11 09:38:24 +02:00
parent 497e27bb9a
commit 3980791cfd
3 changed files with 11 additions and 12 deletions

View File

@@ -82,7 +82,7 @@ def test_query_struct_with_tokens():
assert q.get_tokens(query.TokenRange(0, 2), query.TOKEN_WORD) == []
assert len(q.get_tokens(query.TokenRange(1, 2), query.TOKEN_WORD)) == 2
partials = q.get_partials_list(query.TokenRange(0, 2))
partials = list(q.iter_partials(query.TokenRange(0, 2)))
assert len(partials) == 2
assert [t.token for t in partials] == [1, 2]