test/python/api/search/test_search_country.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) 2025 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Tests for running the country searcher.
   9 """
  10 import pytest
  11
  12 import nominatim_api as napi
  13 from nominatim_api.types import SearchDetails
  14 from nominatim_api.search.db_searches import CountrySearch
  15 from nominatim_api.search.db_search_fields import WeightedStrings
  16
  17
  18 def run_search(apiobj, frontend, global_penalty, ccodes,
  19                country_penalties=None, details=SearchDetails()):
  20     if country_penalties is None:
  21         country_penalties = [0.0] * len(ccodes)
  22
  23     class MySearchData:
  24         penalty = global_penalty
  25         countries = WeightedStrings(ccodes, country_penalties)
  26
  27     search = CountrySearch(MySearchData())
  28     api = frontend(apiobj, options=['search'])
  29
  30     async def run():
  31         async with api._async_api.begin() as conn:
  32             return await search.lookup(conn, details)
  33
  34     return api._loop.run_until_complete(run())
  35
  36
  37 def test_find_from_placex(apiobj, frontend):
  38     apiobj.add_placex(place_id=55, class_='boundary', type='administrative',
  39                       rank_search=4, rank_address=4,
  40                       name={'name': 'Lolaland'},
  41                       country_code='yw',
  42                       centroid=(10, 10),
  43                       geometry='POLYGON((9.5 9.5, 9.5 10.5, 10.5 10.5, 10.5 9.5, 9.5 9.5))')
  44
  45     results = run_search(apiobj, frontend, 0.5, ['de', 'yw'], [0.0, 0.3])
  46
  47     assert len(results) == 1
  48     assert results[0].place_id == 55
  49     assert results[0].accuracy == 0.8
  50
  51
  52 def test_find_from_fallback_countries(apiobj, frontend):
  53     apiobj.add_country('ro', 'POLYGON((0 0, 0 1, 1 1, 1 0, 0 0))')
  54     apiobj.add_country_name('ro', {'name': 'România'})
  55
  56     results = run_search(apiobj, frontend, 0.0, ['ro'])
  57
  58     assert len(results) == 1
  59     assert results[0].names == {'name': 'România'}
  60
  61
  62 def test_find_none(apiobj, frontend):
  63     assert len(run_search(apiobj, frontend, 0.0, ['xx'])) == 0
  64
  65
  66 @pytest.mark.parametrize('coord,numres', [((0.5, 1), 1), ((10, 10), 0)])
  67 def test_find_near(apiobj, frontend, coord, numres):
  68     apiobj.add_country('ro', 'POLYGON((0 0, 0 1, 1 1, 1 0, 0 0))')
  69     apiobj.add_country_name('ro', {'name': 'România'})
  70
  71     results = run_search(apiobj, frontend, 0.0, ['ro'],
  72                          details=SearchDetails(near=napi.Point(*coord),
  73                                                near_radius=0.1))
  74
  75     assert len(results) == numres
  76
  77
  78 class TestCountryParameters:
  79
  80     @pytest.fixture(autouse=True)
  81     def fill_database(self, apiobj):
  82         apiobj.add_placex(place_id=55, class_='boundary', type='administrative',
  83                           rank_search=4, rank_address=4,
  84                           name={'name': 'Lolaland'},
  85                           country_code='yw',
  86                           centroid=(10, 10),
  87                           geometry='POLYGON((9.5 9.5, 9.5 10.5, 10.5 10.5, 10.5 9.5, 9.5 9.5))')
  88         apiobj.add_country('ro', 'POLYGON((0 0, 0 1, 1 1, 1 0, 0 0))')
  89         apiobj.add_country_name('ro', {'name': 'România'})
  90
  91     @pytest.mark.parametrize('geom', [napi.GeometryFormat.GEOJSON,
  92                                       napi.GeometryFormat.KML,
  93                                       napi.GeometryFormat.SVG,
  94                                       napi.GeometryFormat.TEXT])
  95     @pytest.mark.parametrize('cc', ['yw', 'ro'])
  96     def test_return_geometries(self, apiobj, frontend, geom, cc):
  97         results = run_search(apiobj, frontend, 0.5, [cc],
  98                              details=SearchDetails(geometry_output=geom))
  99
 100         assert len(results) == 1
 101         assert geom.name.lower() in results[0].geometry
 102
 103     @pytest.mark.parametrize('pid,rids', [(76, [55]), (55, [])])
 104     def test_exclude_place_id(self, apiobj, frontend, pid, rids):
 105         results = run_search(apiobj, frontend, 0.5, ['yw', 'ro'],
 106                              details=SearchDetails(excluded=[pid]))
 107
 108         assert [r.place_id for r in results] == rids
 109
 110     @pytest.mark.parametrize('viewbox,rids', [((9, 9, 11, 11), [55]),
 111                                               ((-10, -10, -3, -3), [])])
 112     def test_bounded_viewbox_in_placex(self, apiobj, frontend, viewbox, rids):
 113         results = run_search(apiobj, frontend, 0.5, ['yw'],
 114                              details=SearchDetails.from_kwargs({'viewbox': viewbox,
 115                                                                 'bounded_viewbox': True}))
 116
 117         assert [r.place_id for r in results] == rids
 118
 119     @pytest.mark.parametrize('viewbox,numres', [((0, 0, 1, 1), 1),
 120                                                 ((-10, -10, -3, -3), 0)])
 121     def test_bounded_viewbox_in_fallback(self, apiobj, frontend, viewbox, numres):
 122         results = run_search(apiobj, frontend, 0.5, ['ro'],
 123                              details=SearchDetails.from_kwargs({'viewbox': viewbox,
 124                                                                 'bounded_viewbox': True}))
 125
 126         assert len(results) == numres