import pytest_asyncio
from nominatim_api import NominatimAPIAsync
-from nominatim_api.search.query import Phrase, PhraseType, TokenType, BreakType
+from nominatim_api.search.query import Phrase, PhraseType, TokenType
+import nominatim_api.search.query as qmod
import nominatim_api.search.icu_tokenizer as tok
from nominatim_api.logging import set_log_output, get_and_disable
table_factory('word',
definition='word_id INT, word_token TEXT, type TEXT, word TEXT, info JSONB')
- async with NominatimAPIAsync(environ={}) as api:
+ async with NominatimAPIAsync() as api:
async with api.begin() as conn:
yield conn
assert query.num_token_slots() == 2
assert query.nodes[0].starting
assert query.nodes[1].starting
- assert query.nodes[1].btype == BreakType.TOKEN
+ assert query.nodes[1].btype == qmod.BREAK_TOKEN
@pytest.mark.asyncio