X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/4315debff588a5aa67840eda07bcc85d265f923b..ad50016c498098d0f0850c3a9de60623a3bd7e1f:/test/python/api/search/test_db_search_builder.py diff --git a/test/python/api/search/test_db_search_builder.py b/test/python/api/search/test_db_search_builder.py index 68f71298..371a6f02 100644 --- a/test/python/api/search/test_db_search_builder.py +++ b/test/python/api/search/test_db_search_builder.py @@ -9,11 +9,11 @@ Tests for creating abstract searches from token assignments. """ import pytest -from nominatim.api.search.query import Token, TokenRange, BreakType, PhraseType, TokenType, QueryStruct, Phrase -from nominatim.api.search.db_search_builder import SearchBuilder -from nominatim.api.search.token_assignment import TokenAssignment -from nominatim.api.types import SearchDetails -import nominatim.api.search.db_searches as dbs +from nominatim_api.search.query import Token, TokenRange, BreakType, PhraseType, TokenType, QueryStruct, Phrase +from nominatim_api.search.db_search_builder import SearchBuilder +from nominatim_api.search.token_assignment import TokenAssignment +from nominatim_api.types import SearchDetails +import nominatim_api.search.db_searches as dbs class MyToken(Token): def get_category(self): @@ -33,7 +33,7 @@ def make_query(*args): q.add_token(TokenRange(start, end), ttype, MyToken(penalty=0.5 if ttype == TokenType.PARTIAL else 0.0, token=tid, count=1, addr_count=1, - lookup_word=word, is_indexed=True)) + lookup_word=word)) return q @@ -397,14 +397,14 @@ def make_counted_searches(name_part, name_full, address_part, address_full, q.add_node(BreakType.END, PhraseType.NONE) q.add_token(TokenRange(0, 1), TokenType.PARTIAL, - MyToken(0.5, 1, name_part, 1, 'name_part', True)) + MyToken(0.5, 1, name_part, 1, 'name_part')) q.add_token(TokenRange(0, 1), TokenType.WORD, - MyToken(0, 101, name_full, 1, 'name_full', True)) + MyToken(0, 101, name_full, 1, 'name_full')) for i in range(num_address_parts): q.add_token(TokenRange(i + 1, i + 2), TokenType.PARTIAL, - MyToken(0.5, 2, address_part, 1, 'address_part', True)) + MyToken(0.5, 2, address_part, 1, 'address_part')) q.add_token(TokenRange(i + 1, i + 2), TokenType.WORD, - MyToken(0, 102, address_full, 1, 'address_full', True)) + MyToken(0, 102, address_full, 1, 'address_full')) builder = SearchBuilder(q, SearchDetails())