test/python/api/search/test_query.py

   1 # SPDX-License-Identifier: GPL-3.0-or-later
   2 #
   3 # This file is part of Nominatim. (https://nominatim.org)
   4 #
   5 # Copyright (C) 2024 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Test data types for search queries.
   9 """
  10 import pytest
  11
  12 import nominatim_api.search.query as nq
  13
  14 def test_token_range_equal():
  15     assert nq.TokenRange(2, 3) == nq.TokenRange(2, 3)
  16     assert not (nq.TokenRange(2, 3) != nq.TokenRange(2, 3))
  17
  18
  19 @pytest.mark.parametrize('lop,rop', [((1, 2), (3, 4)),
  20                                     ((3, 4), (3, 5)),
  21                                     ((10, 12), (11, 12))])
  22 def test_token_range_unequal(lop, rop):
  23     assert not (nq.TokenRange(*lop) == nq.TokenRange(*rop))
  24     assert nq.TokenRange(*lop) != nq.TokenRange(*rop)
  25
  26
  27 def test_token_range_lt():
  28     assert nq.TokenRange(1, 3) < nq.TokenRange(10, 12)
  29     assert nq.TokenRange(5, 6) < nq.TokenRange(7, 8)
  30     assert nq.TokenRange(1, 4) < nq.TokenRange(4, 5)
  31     assert not(nq.TokenRange(5, 6) < nq.TokenRange(5, 6))
  32     assert not(nq.TokenRange(10, 11) < nq.TokenRange(4, 5))
  33
  34
  35 def test_token_rankge_gt():
  36     assert nq.TokenRange(3, 4) > nq.TokenRange(1, 2)
  37     assert nq.TokenRange(100, 200) > nq.TokenRange(10, 11)
  38     assert nq.TokenRange(10, 11) > nq.TokenRange(4, 10)
  39     assert not(nq.TokenRange(5, 6) > nq.TokenRange(5, 6))
  40     assert not(nq.TokenRange(1, 2) > nq.TokenRange(3, 4))
  41     assert not(nq.TokenRange(4, 10) > nq.TokenRange(3, 5))
  42
  43
  44 def test_token_range_unimplemented_ops():
  45     with pytest.raises(TypeError):
  46         nq.TokenRange(1, 3) <= nq.TokenRange(10, 12)
  47     with pytest.raises(TypeError):
  48         nq.TokenRange(1, 3) >= nq.TokenRange(10, 12)
  49
  50
  51 def test_query_extract_words():
  52     q = nq.QueryStruct([])
  53     q.add_node(nq.BREAK_WORD, nq.PHRASE_ANY, 0.1, '12', '')
  54     q.add_node(nq.BREAK_TOKEN, nq.PHRASE_ANY, 0.0, 'ab', '')
  55     q.add_node(nq.BREAK_PHRASE, nq.PHRASE_ANY, 0.0, '12', '')
  56     q.add_node(nq.BREAK_END, nq.PHRASE_ANY, 0.5, 'hallo', '')
  57
  58     words = q.extract_words(base_penalty=1.0)
  59
  60     assert set(words.keys()) \
  61              == {'12', 'ab', 'hallo', '12 ab', 'ab 12', '12 ab 12'}
  62     assert sorted(words['12']) == [nq.TokenRange(0, 1, 1.0), nq.TokenRange(2, 3, 1.0)]
  63     assert words['12 ab'] == [nq.TokenRange(0, 2, 1.1)]
  64     assert words['hallo'] == [nq.TokenRange(3, 4, 1.0)]
  65