14710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom test.test_support import run_unittest
24710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom _locale import (setlocale, LC_NUMERIC, localeconv, Error)
34710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmtry:
44710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    from _locale import (RADIXCHAR, THOUSEP, nl_langinfo)
54710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmexcept ImportError:
64710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    nl_langinfo = None
74710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
84710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmimport unittest
94710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmimport sys
104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom platform import uname
114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmif uname()[0] == "Darwin":
134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    maj, min, mic = [int(part) for part in uname()[2].split(".")]
144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    if (maj, min, mic) < (8, 0, 0):
154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        raise unittest.SkipTest("locale support broken for OS X < 10.4")
164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmcandidate_locales = ['es_UY', 'fr_FR', 'fi_FI', 'es_CO', 'pt_PT', 'it_IT',
184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'et_EE', 'es_PY', 'no_NO', 'nl_NL', 'lv_LV', 'el_GR', 'be_BY', 'fr_BE',
194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'ro_RO', 'ru_UA', 'ru_RU', 'es_VE', 'ca_ES', 'se_NO', 'es_EC', 'id_ID',
204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'ka_GE', 'es_CL', 'hu_HU', 'wa_BE', 'lt_LT', 'sl_SI', 'hr_HR', 'es_AR',
214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'es_ES', 'oc_FR', 'gl_ES', 'bg_BG', 'is_IS', 'mk_MK', 'de_AT', 'pt_BR',
224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'da_DK', 'nn_NO', 'cs_CZ', 'de_LU', 'es_BO', 'sq_AL', 'sk_SK', 'fr_CH',
234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'de_DE', 'sr_YU', 'br_FR', 'nl_BE', 'sv_FI', 'pl_PL', 'fr_CA', 'fo_FO',
244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'bs_BA', 'fr_LU', 'kl_GL', 'fa_IR', 'de_BE', 'sv_SE', 'it_CH', 'uk_UA',
254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'eu_ES', 'vi_VN', 'af_ZA', 'nb_NO', 'en_DK', 'tg_TJ', 'en_US',
264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    'es_ES.ISO8859-1', 'fr_FR.ISO8859-15', 'ru_RU.KOI8-R', 'ko_KR.eucKR']
274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# Workaround for MSVC6(debug) crash bug
294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmif "MSC v.1200" in sys.version:
304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def accept(loc):
314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        a = loc.split(".")
324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        return not(len(a) == 2 and len(a[-1]) >= 9)
334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    candidate_locales = [loc for loc in candidate_locales if accept(loc)]
344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# List known locale values to test against when available.
364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# Dict formatted as ``<locale> : (<decimal_point>, <thousands_sep>)``.  If a
374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# value is not known, use '' .
384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmknown_numerics = {'fr_FR' : (',', ''), 'en_US':('.', ',')}
394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmclass _LocaleTests(unittest.TestCase):
414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def setUp(self):
434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        self.oldlocale = setlocale(LC_NUMERIC)
444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def tearDown(self):
464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        setlocale(LC_NUMERIC, self.oldlocale)
474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    # Want to know what value was calculated, what it was compared against,
494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    # what function was used for the calculation, what type of data was used,
504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    # the locale that was supposedly set, and the actual locale that is set.
514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    lc_numeric_err_msg = "%s != %s (%s for %s; set to %s, using %s)"
524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def numeric_tester(self, calc_type, calc_value, data_type, used_locale):
544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        """Compare calculation against known value, if available"""
554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        try:
564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            set_locale = setlocale(LC_NUMERIC)
574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        except Error:
584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            set_locale = "<not able to determine>"
594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        known_value = known_numerics.get(used_locale,
604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                    ('', ''))[data_type == 'thousands_sep']
614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        if known_value and calc_value:
624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            self.assertEqual(calc_value, known_value,
634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                self.lc_numeric_err_msg % (
644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                    calc_value, known_value,
654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                    calc_type, data_type, set_locale,
664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                    used_locale))
674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    @unittest.skipUnless(nl_langinfo, "nl_langinfo is not available")
694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def test_lc_numeric_nl_langinfo(self):
704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        # Test nl_langinfo against known values
714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        for loc in candidate_locales:
724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            try:
734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                setlocale(LC_NUMERIC, loc)
744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            except Error:
754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                continue
764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            for li, lc in ((RADIXCHAR, "decimal_point"),
774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                            (THOUSEP, "thousands_sep")):
784710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                self.numeric_tester('nl_langinfo', nl_langinfo(li), lc, loc)
794710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
804710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def test_lc_numeric_localeconv(self):
814710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        # Test localeconv against known values
824710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        for loc in candidate_locales:
834710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            try:
844710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                setlocale(LC_NUMERIC, loc)
854710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            except Error:
864710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                continue
874710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            for lc in ("decimal_point", "thousands_sep"):
884710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                self.numeric_tester('localeconv', localeconv()[lc], lc, loc)
894710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
904710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    @unittest.skipUnless(nl_langinfo, "nl_langinfo is not available")
914710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def test_lc_numeric_basic(self):
924710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        # Test nl_langinfo against localeconv
934710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        for loc in candidate_locales:
944710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            try:
954710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                setlocale(LC_NUMERIC, loc)
964710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            except Error:
974710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                continue
984710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            for li, lc in ((RADIXCHAR, "decimal_point"),
994710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                            (THOUSEP, "thousands_sep")):
1004710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                nl_radixchar = nl_langinfo(li)
1014710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                li_radixchar = localeconv()[lc]
1024710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                try:
1034710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                    set_locale = setlocale(LC_NUMERIC)
1044710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                except Error:
1054710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                    set_locale = "<not able to determine>"
1064710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                self.assertEqual(nl_radixchar, li_radixchar,
1074710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                "%s (nl_langinfo) != %s (localeconv) "
1084710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                "(set to %s, using %s)" % (
1094710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                                nl_radixchar, li_radixchar,
1104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                                loc, set_locale))
1114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
1124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    def test_float_parsing(self):
1134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        # Bug #1391872: Test whether float parsing is okay on European
1144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        # locales.
1154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm        for loc in candidate_locales:
1164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            try:
1174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                setlocale(LC_NUMERIC, loc)
1184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            except Error:
1194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                continue
1204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
1214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            # Ignore buggy locale databases. (Mac OS 10.4 and some other BSDs)
1224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            if loc == 'eu_ES' and localeconv()['decimal_point'] == "' ":
1234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                continue
1244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
1254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            self.assertEqual(int(eval('3.14') * 100), 314,
1264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                "using eval('3.14') failed for %s" % loc)
1274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            self.assertEqual(int(float('3.14') * 100), 314,
1284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                "using float('3.14') failed for %s" % loc)
1294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm            if localeconv()['decimal_point'] != '.':
1304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                self.assertRaises(ValueError, float,
1314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm                                  localeconv()['decimal_point'].join(['1', '23']))
1324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
1334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmdef test_main():
1344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    run_unittest(_LocaleTests)
1354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm
1364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmif __name__ == '__main__':
1374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm    test_main()
138