X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/610f2cc254cf442c895351907be6405f03026903..2171b385519841af0e175d88e51741c5322fba25:/test/python/tokenizer/sanitizers/test_clean_housenumbers.py diff --git a/test/python/tokenizer/sanitizers/test_clean_housenumbers.py b/test/python/tokenizer/sanitizers/test_clean_housenumbers.py index 8a27de6f..11a71a5f 100644 --- a/test/python/tokenizer/sanitizers/test_clean_housenumbers.py +++ b/test/python/tokenizer/sanitizers/test_clean_housenumbers.py @@ -10,17 +10,17 @@ Tests for the sanitizer that normalizes housenumbers. import pytest from nominatim.tokenizer.place_sanitizer import PlaceSanitizer -from nominatim.indexer.place_info import PlaceInfo +from nominatim.data.place_info import PlaceInfo @pytest.fixture -def sanitize(request): +def sanitize(request, def_config): 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) + _, address = PlaceSanitizer([sanitizer_args], def_config).process_names(place) return sorted([(p.kind, p.name) for p in address]) @@ -45,13 +45,24 @@ def test_filter_kind(sanitize): @pytest.mark.parametrize('number', ('6523', 'n/a', '4')) -def test_convert_to_name_converted(number): +def test_convert_to_name_converted(def_config, number): sanitizer_args = {'step': 'clean-housenumbers', 'convert-to-name': (r'\d+', 'n/a')} place = PlaceInfo({'address': {'housenumber': number}}) - names, address = PlaceSanitizer([sanitizer_args]).process_names(place) + names, address = PlaceSanitizer([sanitizer_args], def_config).process_names(place) assert ('housenumber', number) in set((p.kind, p.name) for p in names) assert 'housenumber' not in set(p.kind for p in address) + +@pytest.mark.parametrize('number', ('a54', 'n.a', 'bow')) +def test_convert_to_name_unconverted(def_config, number): + sanitizer_args = {'step': 'clean-housenumbers', + 'convert-to-name': (r'\d+', 'n/a')} + + place = PlaceInfo({'address': {'housenumber': number}}) + names, address = PlaceSanitizer([sanitizer_args], def_config).process_names(place) + + assert 'housenumber' not in set(p.kind for p in names) + assert ('housenumber', number) in set((p.kind, p.name) for p in address)