mirror of
https://github.com/osm-search/Nominatim.git
synced 2026-02-15 02:47:59 +00:00
clean_housenumbers: make kinds and delimiters configurable
Also adds unit tests for various options.
This commit is contained in:
44
test/python/tokenizer/sanitizers/test_clean_housenumbers.py
Normal file
44
test/python/tokenizer/sanitizers/test_clean_housenumbers.py
Normal file
@@ -0,0 +1,44 @@
|
||||
# SPDX-License-Identifier: GPL-2.0-only
|
||||
#
|
||||
# This file is part of Nominatim. (https://nominatim.org)
|
||||
#
|
||||
# Copyright (C) 2022 by the Nominatim developer community.
|
||||
# For a full list of authors see the git log.
|
||||
"""
|
||||
Tests for the sanitizer that normalizes housenumbers.
|
||||
"""
|
||||
import pytest
|
||||
|
||||
from nominatim.tokenizer.place_sanitizer import PlaceSanitizer
|
||||
from nominatim.indexer.place_info import PlaceInfo
|
||||
|
||||
@pytest.fixture
|
||||
def sanitize(request):
|
||||
sanitizer_args = {'step': 'clean-housenumbers'}
|
||||
for mark in request.node.iter_markers(name="sanitizer_params"):
|
||||
sanitizer_args.update({k.replace('_', '-') : v for k,v in mark.kwargs.items()})
|
||||
|
||||
def _run(**kwargs):
|
||||
place = PlaceInfo({'address': kwargs})
|
||||
_, address = PlaceSanitizer([sanitizer_args]).process_names(place)
|
||||
|
||||
return sorted([(p.kind, p.name) for p in address])
|
||||
|
||||
return _run
|
||||
|
||||
|
||||
def test_simple_number(sanitize):
|
||||
assert sanitize(housenumber='34') == [('housenumber', '34')]
|
||||
|
||||
|
||||
@pytest.mark.parametrize('number', ['1;2;3', '1,2,3', '1; 3 ,2',
|
||||
'2,,3,1', '1;2;3;;', ';3;2;1'])
|
||||
def test_housenumber_lists(sanitize, number):
|
||||
assert sanitize(housenumber=number) == \
|
||||
[('housenumber', '1'), ('housenumber', '2'), ('housenumber', '3')]
|
||||
|
||||
|
||||
@pytest.mark.sanitizer_params(filter_kind=('number', 'streetnumber'))
|
||||
def test_filter_kind(sanitize):
|
||||
assert sanitize(housenumber='34', number='4', badnumber='65') == \
|
||||
[('badnumber', '65'), ('housenumber', '34'), ('housenumber', '4')]
|
||||
43
test/python/tokenizer/sanitizers/test_helpers.py
Normal file
43
test/python/tokenizer/sanitizers/test_helpers.py
Normal file
@@ -0,0 +1,43 @@
|
||||
# SPDX-License-Identifier: GPL-2.0-only
|
||||
#
|
||||
# This file is part of Nominatim. (https://nominatim.org)
|
||||
#
|
||||
# Copyright (C) 2022 by the Nominatim developer community.
|
||||
# For a full list of authors see the git log.
|
||||
"""
|
||||
Tests for sanitizer helper functions.
|
||||
"""
|
||||
import pytest
|
||||
|
||||
from nominatim.errors import UsageError
|
||||
import nominatim.tokenizer.sanitizers.helpers as helpers
|
||||
|
||||
@pytest.mark.parametrize('inp', ('fg34', 'f\\f', 'morning [glory]', '56.78'))
|
||||
def test_create_split_regex_no_params_unsplit(inp):
|
||||
regex = helpers.create_split_regex({})
|
||||
|
||||
assert list(regex.split(inp)) == [inp]
|
||||
|
||||
|
||||
@pytest.mark.parametrize('inp,outp', [('here,there', ['here', 'there']),
|
||||
('ying;;yang', ['ying', 'yang']),
|
||||
(';a; ;c;d,', ['', 'a', '', 'c', 'd', '']),
|
||||
('1, 3 ,5', ['1', '3', '5'])
|
||||
])
|
||||
def test_create_split_regex_no_params_split(inp, outp):
|
||||
regex = helpers.create_split_regex({})
|
||||
|
||||
assert list(regex.split(inp)) == outp
|
||||
|
||||
|
||||
@pytest.mark.parametrize('delimiter', ['.', '\\', '[]', ' ', '/.*+'])
|
||||
def test_create_split_regex_custom(delimiter):
|
||||
regex = helpers.create_split_regex({'delimiters': delimiter})
|
||||
|
||||
assert list(regex.split(f'out{delimiter}house')) == ['out', 'house']
|
||||
assert list(regex.split('out,house')) == ['out,house']
|
||||
|
||||
|
||||
def test_create_split_regex_empty_delimiter():
|
||||
with pytest.raises(UsageError):
|
||||
regex = helpers.create_split_regex({'delimiters': ''})
|
||||
@@ -5,7 +5,7 @@
|
||||
# Copyright (C) 2022 by the Nominatim developer community.
|
||||
# For a full list of authors see the git log.
|
||||
"""
|
||||
Tests for the sanitizer that splitts multivalue lists.
|
||||
Tests for the sanitizer that splits multivalue lists.
|
||||
"""
|
||||
import pytest
|
||||
|
||||
|
||||
Reference in New Issue
Block a user