183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehfrom test.test_support import run_unittest
283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehfrom _locale import (setlocale, LC_NUMERIC, localeconv, Error)
383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehtry:
483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    from _locale import (RADIXCHAR, THOUSEP, nl_langinfo)
583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehexcept ImportError:
683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    nl_langinfo = None
783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehimport unittest
983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehimport sys
1083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehfrom platform import uname
1183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
1283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehif uname()[0] == "Darwin":
1383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    maj, min, mic = [int(part) for part in uname()[2].split(".")]
1483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    if (maj, min, mic) < (8, 0, 0):
1583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        raise unittest.SkipTest("locale support broken for OS X < 10.4")
1683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
1783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehcandidate_locales = ['es_UY', 'fr_FR', 'fi_FI', 'es_CO', 'pt_PT', 'it_IT',
1883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'et_EE', 'es_PY', 'no_NO', 'nl_NL', 'lv_LV', 'el_GR', 'be_BY', 'fr_BE',
1983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'ro_RO', 'ru_UA', 'ru_RU', 'es_VE', 'ca_ES', 'se_NO', 'es_EC', 'id_ID',
2083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'ka_GE', 'es_CL', 'hu_HU', 'wa_BE', 'lt_LT', 'sl_SI', 'hr_HR', 'es_AR',
2183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'es_ES', 'oc_FR', 'gl_ES', 'bg_BG', 'is_IS', 'mk_MK', 'de_AT', 'pt_BR',
2283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'da_DK', 'nn_NO', 'cs_CZ', 'de_LU', 'es_BO', 'sq_AL', 'sk_SK', 'fr_CH',
2383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'de_DE', 'sr_YU', 'br_FR', 'nl_BE', 'sv_FI', 'pl_PL', 'fr_CA', 'fo_FO',
2483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'bs_BA', 'fr_LU', 'kl_GL', 'fa_IR', 'de_BE', 'sv_SE', 'it_CH', 'uk_UA',
2583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'eu_ES', 'vi_VN', 'af_ZA', 'nb_NO', 'en_DK', 'tg_TJ', 'en_US',
2683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    'es_ES.ISO8859-1', 'fr_FR.ISO8859-15', 'ru_RU.KOI8-R', 'ko_KR.eucKR']
2783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
2883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh# Workaround for MSVC6(debug) crash bug
2983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehif "MSC v.1200" in sys.version:
3083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def accept(loc):
3183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        a = loc.split(".")
3283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        return not(len(a) == 2 and len(a[-1]) >= 9)
3383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    candidate_locales = [loc for loc in candidate_locales if accept(loc)]
3483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
3583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh# List known locale values to test against when available.
3683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh# Dict formatted as ``<locale> : (<decimal_point>, <thousands_sep>)``.  If a
3783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh# value is not known, use '' .
3883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehknown_numerics = {'fr_FR' : (',', ''), 'en_US':('.', ',')}
3983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
4083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehclass _LocaleTests(unittest.TestCase):
4183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
4283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def setUp(self):
4383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        self.oldlocale = setlocale(LC_NUMERIC)
4483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
4583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def tearDown(self):
4683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        setlocale(LC_NUMERIC, self.oldlocale)
4783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
4883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    # Want to know what value was calculated, what it was compared against,
4983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    # what function was used for the calculation, what type of data was used,
5083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    # the locale that was supposedly set, and the actual locale that is set.
5183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    lc_numeric_err_msg = "%s != %s (%s for %s; set to %s, using %s)"
5283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
5383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def numeric_tester(self, calc_type, calc_value, data_type, used_locale):
5483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        """Compare calculation against known value, if available"""
5583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        try:
5683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            set_locale = setlocale(LC_NUMERIC)
5783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        except Error:
5883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            set_locale = "<not able to determine>"
5983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        known_value = known_numerics.get(used_locale,
6083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                    ('', ''))[data_type == 'thousands_sep']
6183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        if known_value and calc_value:
6283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            self.assertEqual(calc_value, known_value,
6383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                self.lc_numeric_err_msg % (
6483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                    calc_value, known_value,
6583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                    calc_type, data_type, set_locale,
6683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                    used_locale))
6783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
6883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    @unittest.skipUnless(nl_langinfo, "nl_langinfo is not available")
6983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def test_lc_numeric_nl_langinfo(self):
7083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        # Test nl_langinfo against known values
7183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        for loc in candidate_locales:
7283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            try:
7383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                setlocale(LC_NUMERIC, loc)
7483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            except Error:
7583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                continue
7683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            for li, lc in ((RADIXCHAR, "decimal_point"),
7783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                            (THOUSEP, "thousands_sep")):
7883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                self.numeric_tester('nl_langinfo', nl_langinfo(li), lc, loc)
7983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
8083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def test_lc_numeric_localeconv(self):
8183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        # Test localeconv against known values
8283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        for loc in candidate_locales:
8383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            try:
8483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                setlocale(LC_NUMERIC, loc)
8583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            except Error:
8683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                continue
8783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            for lc in ("decimal_point", "thousands_sep"):
8883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                self.numeric_tester('localeconv', localeconv()[lc], lc, loc)
8983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
9083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    @unittest.skipUnless(nl_langinfo, "nl_langinfo is not available")
9183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def test_lc_numeric_basic(self):
9283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        # Test nl_langinfo against localeconv
9383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        for loc in candidate_locales:
9483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            try:
9583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                setlocale(LC_NUMERIC, loc)
9683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            except Error:
9783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                continue
9883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            for li, lc in ((RADIXCHAR, "decimal_point"),
9983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                            (THOUSEP, "thousands_sep")):
10083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                nl_radixchar = nl_langinfo(li)
10183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                li_radixchar = localeconv()[lc]
10283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                try:
10383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                    set_locale = setlocale(LC_NUMERIC)
10483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                except Error:
10583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                    set_locale = "<not able to determine>"
10683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                self.assertEqual(nl_radixchar, li_radixchar,
10783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                "%s (nl_langinfo) != %s (localeconv) "
10883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                "(set to %s, using %s)" % (
10983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                                nl_radixchar, li_radixchar,
11083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                                loc, set_locale))
11183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
11283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    def test_float_parsing(self):
11383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        # Bug #1391872: Test whether float parsing is okay on European
11483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        # locales.
11583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh        for loc in candidate_locales:
11683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            try:
11783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                setlocale(LC_NUMERIC, loc)
11883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            except Error:
11983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                continue
12083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
12183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            # Ignore buggy locale databases. (Mac OS 10.4 and some other BSDs)
12283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            if loc == 'eu_ES' and localeconv()['decimal_point'] == "' ":
12383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                continue
12483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
12583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            self.assertEqual(int(eval('3.14') * 100), 314,
12683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                "using eval('3.14') failed for %s" % loc)
12783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            self.assertEqual(int(float('3.14') * 100), 314,
12883760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                "using float('3.14') failed for %s" % loc)
12983760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh            if localeconv()['decimal_point'] != '.':
13083760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                self.assertRaises(ValueError, float,
13183760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh                                  localeconv()['decimal_point'].join(['1', '23']))
13283760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
13383760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehdef test_main():
13483760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    run_unittest(_LocaleTests)
13583760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh
13683760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsiehif __name__ == '__main__':
13783760d213fb3bec7b4117d266fcfbf6fe2ba14abAndrew Hsieh    test_main()
138