X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/3bf3b894eaddd4f17f7e92353af6b2aa6200ab20..b8a73192128b162ea32bf2f622875a543a39d955:/test/python/tokenizer/token_analysis/test_analysis_postcodes.py diff --git a/test/python/tokenizer/token_analysis/test_analysis_postcodes.py b/test/python/tokenizer/token_analysis/test_analysis_postcodes.py index 623bed54..8d966c46 100644 --- a/test/python/tokenizer/token_analysis/test_analysis_postcodes.py +++ b/test/python/tokenizer/token_analysis/test_analysis_postcodes.py @@ -12,6 +12,7 @@ import pytest from icu import Transliterator import nominatim.tokenizer.token_analysis.postcodes as module +from nominatim.data.place_name import PlaceName from nominatim.errors import UsageError DEFAULT_NORMALIZATION = """ :: NFD (); @@ -39,22 +40,22 @@ def analyser(): def get_normalized_variants(proc, name): norm = Transliterator.createFromRules("test_norm", DEFAULT_NORMALIZATION) - return proc.get_variants_ascii(norm.transliterate(name).strip()) + return proc.compute_variants(norm.transliterate(name).strip()) @pytest.mark.parametrize('name,norm', [('12', '12'), ('A 34 ', 'A 34'), ('34-av', '34-AV')]) -def test_normalize(analyser, name, norm): - assert analyser.normalize(name) == norm +def test_get_canonical_id(analyser, name, norm): + assert analyser.get_canonical_id(PlaceName(name=name, kind='', suffix='')) == norm @pytest.mark.parametrize('postcode,variants', [('12345', {'12345'}), ('AB-998', {'ab 998', 'ab998'}), ('23 FGH D3', {'23 fgh d3', '23fgh d3', '23 fghd3', '23fghd3'})]) -def test_get_variants_ascii(analyser, postcode, variants): - out = analyser.get_variants_ascii(postcode) +def test_compute_variants(analyser, postcode, variants): + out = analyser.compute_variants(postcode) assert len(out) == len(set(out)) assert set(out) == variants