test/python/api/test_localization.py

   1 # SPDX-License-Identifier: GPL-3.0-or-later
   2 #
   3 # This file is part of Nominatim. (https://nominatim.org)
   4 #
   5 # Copyright (C) 2025 by the Nominatim developer community.
   6 # For a full list of authors see the git log.
   7 """
   8 Test functions for adapting results to the user's locale.
   9 """
  10 import pytest
  11
  12 from nominatim_api import Locales
  13
  14
  15 def test_display_name_empty_names():
  16     loc = Locales(['en', 'de'])
  17
  18     assert loc.display_name(None) == ''
  19     assert loc.display_name({}) == ''
  20
  21
  22 def test_display_name_none_localized():
  23     loc = Locales()
  24
  25     assert loc.display_name({}) == ''
  26     assert loc.display_name({'name:de': 'DE', 'name': 'ALL'}) == 'ALL'
  27     assert loc.display_name({'ref': '34', 'name:de': 'DE'}) == '34'
  28
  29
  30 def test_display_name_localized():
  31     loc = Locales(['en', 'de'])
  32
  33     assert loc.display_name({}) == ''
  34     assert loc.display_name({'name:de': 'DE', 'name': 'ALL'}) == 'DE'
  35     assert loc.display_name({'ref': '34', 'name:de': 'DE'}) == 'DE'
  36
  37
  38 def test_display_name_preference():
  39     loc = Locales(['en', 'de'])
  40
  41     assert loc.display_name({}) == ''
  42     assert loc.display_name({'name:de': 'DE', 'name:en': 'EN'}) == 'EN'
  43     assert loc.display_name({'official_name:en': 'EN', 'name:de': 'DE'}) == 'DE'
  44
  45
  46 @pytest.mark.parametrize('langstr,langlist',
  47                          [('fr', ['fr']),
  48                           ('fr-FR', ['fr-FR', 'fr']),
  49                           ('de,fr-FR', ['de', 'fr-FR', 'fr']),
  50                           ('fr,de,fr-FR', ['fr', 'de', 'fr-FR']),
  51                           ('en;q=0.5,fr', ['fr', 'en']),
  52                           ('en;q=0.5,fr,en-US', ['fr', 'en-US', 'en']),
  53                           ('en,fr;garbage,de', ['en', 'de'])])
  54 def test_from_language_preferences(langstr, langlist):
  55     assert Locales.from_accept_languages(langstr).languages == langlist