test/python/api/test_api_search.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 Tests for search API calls.
   9
  10 These tests make sure that all Python code is correct and executable.
  11 Functional tests can be found in the BDD test suite.
  12 """
  13 import json
  14
  15 import pytest
  16
  17 import sqlalchemy as sa
  18
  19 import nominatim_api as napi
  20 import nominatim_api.logging as loglib
  21
  22 API_OPTIONS = {'search'}
  23
  24 @pytest.fixture(autouse=True)
  25 def setup_icu_tokenizer(apiobj):
  26     """ Setup the properties needed for using the ICU tokenizer.
  27     """
  28     apiobj.add_data('properties',
  29                     [{'property': 'tokenizer', 'value': 'icu'},
  30                      {'property': 'tokenizer_import_normalisation', 'value': ':: lower();'},
  31                      {'property': 'tokenizer_import_transliteration', 'value': "'1' > '/1/'; 'ä' > 'ä '"},
  32                     ])
  33
  34
  35 def test_search_no_content(apiobj, frontend):
  36     apiobj.add_word_table([])
  37
  38     api = frontend(apiobj, options=API_OPTIONS)
  39     assert api.search('foo') == []
  40
  41
  42 def test_search_simple_word(apiobj, frontend):
  43     apiobj.add_word_table([(55, 'test', 'W', 'test', None),
  44                            (2, 'test', 'w', 'test', None)])
  45
  46     apiobj.add_placex(place_id=444, class_='place', type='village',
  47                       centroid=(1.3, 0.7))
  48     apiobj.add_search_name(444, names=[2, 55])
  49
  50     api = frontend(apiobj, options=API_OPTIONS)
  51     results = api.search('TEST')
  52
  53     assert [r.place_id for r in results] == [444]
  54
  55
  56 @pytest.mark.parametrize('logtype', ['text', 'html'])
  57 def test_search_with_debug(apiobj, frontend, logtype):
  58     apiobj.add_word_table([(55, 'test', 'W', 'test', None),
  59                            (2, 'test', 'w', 'test', None)])
  60
  61     apiobj.add_placex(place_id=444, class_='place', type='village',
  62                       centroid=(1.3, 0.7))
  63     apiobj.add_search_name(444, names=[2, 55])
  64
  65     api = frontend(apiobj, options=API_OPTIONS)
  66     loglib.set_log_output(logtype)
  67     results = api.search('TEST')
  68
  69     assert loglib.get_and_disable()
  70
  71
  72 def test_address_no_content(apiobj, frontend):
  73     apiobj.add_word_table([])
  74
  75     api = frontend(apiobj, options=API_OPTIONS)
  76     assert api.search_address(amenity='hotel',
  77                               street='Main St 34',
  78                               city='Happyville',
  79                               county='Wideland',
  80                               state='Praerie',
  81                               postalcode='55648',
  82                               country='xx') == []
  83
  84
  85 @pytest.mark.parametrize('atype,address,search', [('street', 26, 26),
  86                                                   ('city', 16, 18),
  87                                                   ('county', 12, 12),
  88                                                   ('state', 8, 8)])
  89 def test_address_simple_places(apiobj, frontend, atype, address, search):
  90     apiobj.add_word_table([(55, 'test', 'W', 'test', None),
  91                            (2, 'test', 'w', 'test', None)])
  92
  93     apiobj.add_placex(place_id=444,
  94                       rank_address=address, rank_search=search,
  95                       centroid=(1.3, 0.7))
  96     apiobj.add_search_name(444, names=[2, 55], address_rank=address, search_rank=search)
  97
  98     api = frontend(apiobj, options=API_OPTIONS)
  99     results = api.search_address(**{atype: 'TEST'})
 100
 101     assert [r.place_id for r in results] == [444]
 102
 103
 104 def test_address_country(apiobj, frontend):
 105     apiobj.add_word_table([(None, 'ro', 'C', 'ro', None)])
 106     apiobj.add_country('ro', 'POLYGON((0 0, 0 1, 1 1, 1 0, 0 0))')
 107     apiobj.add_country_name('ro', {'name': 'România'})
 108
 109     api = frontend(apiobj, options=API_OPTIONS)
 110     assert len(api.search_address(country='ro')) == 1
 111
 112
 113 def test_category_no_categories(apiobj, frontend):
 114     apiobj.add_word_table([])
 115
 116     api = frontend(apiobj, options=API_OPTIONS)
 117     assert api.search_category([], near_query='Berlin') == []
 118
 119
 120 def test_category_no_content(apiobj, frontend):
 121     apiobj.add_word_table([])
 122
 123     api = frontend(apiobj, options=API_OPTIONS)
 124     assert api.search_category([('amenity', 'restaurant')]) == []
 125
 126
 127 def test_category_simple_restaurant(apiobj, frontend):
 128     apiobj.add_word_table([])
 129
 130     apiobj.add_placex(place_id=444, class_='amenity', type='restaurant',
 131                       centroid=(1.3, 0.7))
 132     apiobj.add_search_name(444, names=[2, 55], address_rank=16, search_rank=18)
 133
 134     api = frontend(apiobj, options=API_OPTIONS)
 135     results = api.search_category([('amenity', 'restaurant')],
 136                                   near=(1.3, 0.701), near_radius=0.015)
 137
 138     assert [r.place_id for r in results] == [444]
 139
 140
 141 def test_category_with_search_phrase(apiobj, frontend):
 142     apiobj.add_word_table([(55, 'test', 'W', 'test', None),
 143                            (2, 'test', 'w', 'test', None)])
 144
 145     apiobj.add_placex(place_id=444, class_='place', type='village',
 146                       rank_address=16, rank_search=18,
 147                       centroid=(1.3, 0.7))
 148     apiobj.add_search_name(444, names=[2, 55], address_rank=16, search_rank=18)
 149     apiobj.add_placex(place_id=95, class_='amenity', type='restaurant',
 150                       centroid=(1.3, 0.7003))
 151
 152     api = frontend(apiobj, options=API_OPTIONS)
 153     results = api.search_category([('amenity', 'restaurant')], near_query='TEST')
 154
 155     assert [r.place_id for r in results] == [95]