forked from hans/Nominatim
00e3a752c9
Use adapted types for the different result types. This makes it easier to have adapted output formatting and means there are only result fields that are filled.
85 lines
2.7 KiB
Python
85 lines
2.7 KiB
Python
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
# This file is part of Nominatim. (https://nominatim.org)
|
|
#
|
|
# Copyright (C) 2023 by the Nominatim developer community.
|
|
# For a full list of authors see the git log.
|
|
"""
|
|
Tests for result datatype helper functions.
|
|
"""
|
|
import struct
|
|
|
|
import pytest
|
|
import pytest_asyncio
|
|
import sqlalchemy as sa
|
|
|
|
|
|
from nominatim.api import SourceTable, DetailedResult, Point
|
|
import nominatim.api.results as nresults
|
|
|
|
class FakeCentroid:
|
|
def __init__(self, x, y):
|
|
self.data = struct.pack("=biidd", 1, 0x20000001, 4326,
|
|
x, y)
|
|
|
|
class FakeRow:
|
|
def __init__(self, **kwargs):
|
|
for k, v in kwargs.items():
|
|
setattr(self, k, v)
|
|
self._mapping = kwargs
|
|
|
|
|
|
def test_minimal_detailed_result():
|
|
res = DetailedResult(SourceTable.PLACEX,
|
|
('amenity', 'post_box'),
|
|
Point(23.1, 0.5))
|
|
|
|
assert res.lon == 23.1
|
|
assert res.lat == 0.5
|
|
assert res.calculated_importance() == pytest.approx(0.0000001)
|
|
|
|
def test_detailed_result_custom_importance():
|
|
res = DetailedResult(SourceTable.PLACEX,
|
|
('amenity', 'post_box'),
|
|
Point(23.1, 0.5),
|
|
importance=0.4563)
|
|
|
|
assert res.calculated_importance() == 0.4563
|
|
|
|
|
|
@pytest.mark.parametrize('func', (nresults.create_from_placex_row,
|
|
nresults.create_from_osmline_row,
|
|
nresults.create_from_tiger_row,
|
|
nresults.create_from_postcode_row))
|
|
def test_create_row_none(func):
|
|
assert func(None, DetailedResult) is None
|
|
|
|
|
|
@pytest.mark.parametrize('func', (nresults.create_from_osmline_row,
|
|
nresults.create_from_tiger_row))
|
|
def test_create_row_with_housenumber(func):
|
|
row = FakeRow(place_id = 2345, osm_id = 111, housenumber = 4,
|
|
address = None, postcode = '99900', country_code = 'xd',
|
|
centroid = FakeCentroid(0, 0))
|
|
|
|
res = func(row, DetailedResult)
|
|
|
|
assert res.housenumber == '4'
|
|
assert res.extratags is None
|
|
assert res.category == ('place', 'house')
|
|
|
|
|
|
@pytest.mark.parametrize('func', (nresults.create_from_osmline_row,
|
|
nresults.create_from_tiger_row))
|
|
def test_create_row_without_housenumber(func):
|
|
row = FakeRow(place_id=2345, osm_id=111,
|
|
startnumber=1, endnumber=11, step=2,
|
|
address=None, postcode='99900', country_code='xd',
|
|
centroid=FakeCentroid(0, 0))
|
|
|
|
res = func(row, DetailedResult)
|
|
|
|
assert res.housenumber is None
|
|
assert res.extratags == {'startnumber': '1', 'endnumber': '11', 'step': '2'}
|
|
assert res.category == ('place', 'houses')
|