mirror of
https://github.com/osm-search/Nominatim.git
synced 2026-02-15 10:57:58 +00:00
288 lines
13 KiB
Python
288 lines
13 KiB
Python
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
# This file is part of Nominatim. (https://nominatim.org)
|
|
#
|
|
# Copyright (C) 2025 by the Nominatim developer community.
|
|
# For a full list of authors see the git log.
|
|
"""
|
|
Tests for running the named place searcher.
|
|
"""
|
|
import json
|
|
|
|
import pytest
|
|
|
|
import nominatim_api as napi
|
|
from nominatim_api.types import SearchDetails
|
|
from nominatim_api.search.db_searches import PlaceSearch
|
|
from nominatim_api.search.db_search_fields import WeightedStrings, WeightedCategories, \
|
|
FieldLookup, FieldRanking, RankedTokens
|
|
from nominatim_api.search.db_search_lookups import LookupAll, LookupAny, Restrict
|
|
|
|
APIOPTIONS = ['search']
|
|
|
|
|
|
def run_search(apiobj, frontend, global_penalty, lookup, ranking, count=2,
|
|
pcs=[], ccodes=[], quals=[], has_address=False,
|
|
details=SearchDetails()):
|
|
class MySearchData:
|
|
penalty = global_penalty
|
|
postcodes = WeightedStrings(pcs, [0.0] * len(pcs))
|
|
countries = WeightedStrings(ccodes, [0.0] * len(ccodes))
|
|
qualifiers = WeightedCategories(quals, [0.0] * len(quals))
|
|
lookups = lookup
|
|
rankings = ranking
|
|
housenumbers = None
|
|
|
|
search = PlaceSearch(0.0, MySearchData(), count, has_address)
|
|
|
|
if frontend is None:
|
|
api = apiobj
|
|
else:
|
|
api = frontend(apiobj, options=APIOPTIONS)
|
|
|
|
async def run():
|
|
async with api._async_api.begin() as conn:
|
|
return await search.lookup(conn, details)
|
|
|
|
results = api._loop.run_until_complete(run())
|
|
results.sort(key=lambda r: r.accuracy)
|
|
|
|
return results
|
|
|
|
|
|
class TestNameOnlySearches:
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def fill_database(self, apiobj):
|
|
apiobj.add_placex(place_id=100, country_code='us',
|
|
centroid=(5.6, 4.3))
|
|
apiobj.add_search_name(100, names=[1, 2, 10, 11], country_code='us',
|
|
centroid=(5.6, 4.3))
|
|
apiobj.add_placex(place_id=101, country_code='mx',
|
|
centroid=(-10.3, 56.9))
|
|
apiobj.add_search_name(101, names=[1, 2, 20, 21], country_code='mx',
|
|
centroid=(-10.3, 56.9))
|
|
|
|
@pytest.mark.parametrize('lookup_type', [LookupAll, Restrict])
|
|
@pytest.mark.parametrize('rank,res', [([10], [100, 101]),
|
|
([20], [101, 100])])
|
|
def test_lookup_all_match(self, apiobj, frontend, lookup_type, rank, res):
|
|
lookup = FieldLookup('name_vector', [1, 2], lookup_type)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, rank)])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking])
|
|
|
|
assert [r.place_id for r in results] == res
|
|
|
|
@pytest.mark.parametrize('lookup_type', [LookupAll, Restrict])
|
|
def test_lookup_all_partial_match(self, apiobj, frontend, lookup_type):
|
|
lookup = FieldLookup('name_vector', [1, 20], lookup_type)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, [21])])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking])
|
|
|
|
assert len(results) == 1
|
|
assert results[0].place_id == 101
|
|
|
|
@pytest.mark.parametrize('rank,res', [([10], [100, 101]),
|
|
([20], [101, 100])])
|
|
def test_lookup_any_match(self, apiobj, frontend, rank, res):
|
|
lookup = FieldLookup('name_vector', [11, 21], LookupAny)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, rank)])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking])
|
|
|
|
assert [r.place_id for r in results] == res
|
|
|
|
def test_lookup_any_partial_match(self, apiobj, frontend):
|
|
lookup = FieldLookup('name_vector', [20], LookupAll)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, [21])])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking])
|
|
|
|
assert len(results) == 1
|
|
assert results[0].place_id == 101
|
|
|
|
@pytest.mark.parametrize('cc,res', [('us', 100), ('mx', 101)])
|
|
def test_lookup_restrict_country(self, apiobj, frontend, cc, res):
|
|
lookup = FieldLookup('name_vector', [1, 2], LookupAll)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, [10])])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking], ccodes=[cc])
|
|
|
|
assert [r.place_id for r in results] == [res]
|
|
|
|
def test_lookup_restrict_placeid(self, apiobj, frontend):
|
|
lookup = FieldLookup('name_vector', [1, 2], LookupAll)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, [10])])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking],
|
|
details=SearchDetails(excluded=[101]))
|
|
|
|
assert [r.place_id for r in results] == [100]
|
|
|
|
@pytest.mark.parametrize('geom', [napi.GeometryFormat.GEOJSON,
|
|
napi.GeometryFormat.KML,
|
|
napi.GeometryFormat.SVG,
|
|
napi.GeometryFormat.TEXT])
|
|
def test_return_geometries(self, apiobj, frontend, geom):
|
|
lookup = FieldLookup('name_vector', [20], LookupAll)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, [21])])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking],
|
|
details=SearchDetails(geometry_output=geom))
|
|
|
|
assert geom.name.lower() in results[0].geometry
|
|
|
|
@pytest.mark.parametrize('factor,npoints', [(0.0, 3), (1.0, 2)])
|
|
def test_return_simplified_geometry(self, apiobj, frontend, factor, npoints):
|
|
apiobj.add_placex(place_id=333, country_code='us',
|
|
centroid=(9.0, 9.0),
|
|
geometry='LINESTRING(8.9 9.0, 9.0 9.0, 9.1 9.0)')
|
|
apiobj.add_search_name(333, names=[55], country_code='us',
|
|
centroid=(5.6, 4.3))
|
|
|
|
lookup = FieldLookup('name_vector', [55], LookupAll)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, [21])])
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [ranking],
|
|
details=SearchDetails(geometry_output=napi.GeometryFormat.GEOJSON,
|
|
geometry_simplification=factor))
|
|
|
|
assert len(results) == 1
|
|
result = results[0]
|
|
geom = json.loads(result.geometry['geojson'])
|
|
|
|
assert result.place_id == 333
|
|
assert len(geom['coordinates']) == npoints
|
|
|
|
@pytest.mark.parametrize('viewbox', ['5.0,4.0,6.0,5.0', '5.7,4.0,6.0,5.0'])
|
|
@pytest.mark.parametrize('wcount,rids', [(2, [100, 101]), (20000, [100])])
|
|
def test_prefer_viewbox(self, apiobj, frontend, viewbox, wcount, rids):
|
|
lookup = FieldLookup('name_vector', [1, 2], LookupAll)
|
|
ranking = FieldRanking('name_vector', 0.2, [RankedTokens(0.0, [21])])
|
|
|
|
api = frontend(apiobj, options=APIOPTIONS)
|
|
results = run_search(api, None, 0.1, [lookup], [ranking])
|
|
assert [r.place_id for r in results] == [101, 100]
|
|
|
|
results = run_search(api, None, 0.1, [lookup], [ranking], count=wcount,
|
|
details=SearchDetails.from_kwargs({'viewbox': viewbox}))
|
|
assert [r.place_id for r in results] == rids
|
|
|
|
@pytest.mark.parametrize('viewbox', ['5.0,4.0,6.0,5.0', '5.55,4.27,5.62,4.31'])
|
|
def test_force_viewbox(self, apiobj, frontend, viewbox):
|
|
lookup = FieldLookup('name_vector', [1, 2], LookupAll)
|
|
|
|
details = SearchDetails.from_kwargs({'viewbox': viewbox,
|
|
'bounded_viewbox': True})
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [], details=details)
|
|
assert [r.place_id for r in results] == [100]
|
|
|
|
def test_prefer_near(self, apiobj, frontend):
|
|
lookup = FieldLookup('name_vector', [1, 2], LookupAll)
|
|
ranking = FieldRanking('name_vector', 0.4, [RankedTokens(0.0, [21])])
|
|
|
|
api = frontend(apiobj, options=APIOPTIONS)
|
|
results = run_search(api, None, 0.1, [lookup], [ranking])
|
|
assert [r.place_id for r in results] == [101, 100]
|
|
|
|
results = run_search(api, None, 0.1, [lookup], [ranking],
|
|
details=SearchDetails.from_kwargs({'near': '5.6,4.3'}))
|
|
results.sort(key=lambda r: -r.importance)
|
|
assert [r.place_id for r in results] == [100, 101]
|
|
|
|
@pytest.mark.parametrize('radius', [0.09, 0.11])
|
|
def test_force_near(self, apiobj, frontend, radius):
|
|
lookup = FieldLookup('name_vector', [1, 2], LookupAll)
|
|
|
|
details = SearchDetails.from_kwargs({'near': '5.6,4.3',
|
|
'near_radius': radius})
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [], details=details)
|
|
|
|
assert [r.place_id for r in results] == [100]
|
|
|
|
|
|
@pytest.mark.parametrize('wcount,rids', [(2, [990, 991]), (30000, [990])])
|
|
def test_name_and_postcode(apiobj, frontend, wcount, rids):
|
|
apiobj.add_placex(place_id=990, class_='highway', type='service',
|
|
rank_search=27, rank_address=27,
|
|
postcode='11225',
|
|
centroid=(10.0, 10.0),
|
|
geometry='LINESTRING(9.995 10, 10.005 10)')
|
|
apiobj.add_search_name(990, names=[111], centroid=(10.0, 10.0),
|
|
search_rank=27, address_rank=27)
|
|
apiobj.add_placex(place_id=991, class_='highway', type='service',
|
|
rank_search=27, rank_address=27,
|
|
postcode='11221',
|
|
centroid=(10.3, 10.3),
|
|
geometry='LINESTRING(9.995 10.3, 10.005 10.3)')
|
|
apiobj.add_search_name(991, names=[111], centroid=(10.3, 10.3),
|
|
search_rank=27, address_rank=27)
|
|
apiobj.add_postcode(place_id=100, country_code='ch', postcode='11225',
|
|
geometry='POINT(10 10)')
|
|
|
|
lookup = FieldLookup('name_vector', [111], LookupAll)
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [], pcs=['11225'], count=wcount,
|
|
details=SearchDetails())
|
|
|
|
assert results
|
|
assert [r.place_id for r in results] == rids
|
|
|
|
|
|
class TestLayersRank30:
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def fill_database(self, apiobj):
|
|
apiobj.add_placex(place_id=223, class_='place', type='house',
|
|
housenumber='1',
|
|
rank_address=30,
|
|
rank_search=30)
|
|
apiobj.add_search_name(223, names=[34],
|
|
importance=0.0009,
|
|
address_rank=30, search_rank=30)
|
|
apiobj.add_placex(place_id=224, class_='amenity', type='toilet',
|
|
rank_address=30,
|
|
rank_search=30)
|
|
apiobj.add_search_name(224, names=[34],
|
|
importance=0.0008,
|
|
address_rank=30, search_rank=30)
|
|
apiobj.add_placex(place_id=225, class_='man_made', type='tower',
|
|
rank_address=0,
|
|
rank_search=30)
|
|
apiobj.add_search_name(225, names=[34],
|
|
importance=0.0007,
|
|
address_rank=0, search_rank=30)
|
|
apiobj.add_placex(place_id=226, class_='railway', type='station',
|
|
rank_address=0,
|
|
rank_search=30)
|
|
apiobj.add_search_name(226, names=[34],
|
|
importance=0.0006,
|
|
address_rank=0, search_rank=30)
|
|
apiobj.add_placex(place_id=227, class_='natural', type='cave',
|
|
rank_address=0,
|
|
rank_search=30)
|
|
apiobj.add_search_name(227, names=[34],
|
|
importance=0.0005,
|
|
address_rank=0, search_rank=30)
|
|
|
|
@pytest.mark.parametrize('layer,res',
|
|
[(napi.DataLayer.ADDRESS, [223]),
|
|
(napi.DataLayer.POI, [224]),
|
|
(napi.DataLayer.ADDRESS | napi.DataLayer.POI, [223, 224]),
|
|
(napi.DataLayer.MANMADE, [225]),
|
|
(napi.DataLayer.RAILWAY, [226]),
|
|
(napi.DataLayer.NATURAL, [227]),
|
|
(napi.DataLayer.MANMADE | napi.DataLayer.NATURAL, [225, 227]),
|
|
(napi.DataLayer.MANMADE | napi.DataLayer.RAILWAY, [225, 226])])
|
|
def test_layers_rank30(self, apiobj, frontend, layer, res):
|
|
lookup = FieldLookup('name_vector', [34], LookupAny)
|
|
|
|
results = run_search(apiobj, frontend, 0.1, [lookup], [],
|
|
details=SearchDetails(layers=layer))
|
|
|
|
assert [r.place_id for r in results] == res
|