| from test.test_support import run_unittest, verbose | |
| import unittest | |
| import locale | |
| import sys | |
| import codecs | |
| enUS_locale = None | |
| def get_enUS_locale(): | |
| global enUS_locale | |
| if sys.platform == 'darwin': | |
| import os | |
| tlocs = ("en_US.UTF-8", "en_US.ISO8859-1", "en_US") | |
| if int(os.uname()[2].split('.')[0]) < 10: | |
| # The locale test work fine on OSX 10.6, I (ronaldoussoren) | |
| # haven't had time yet to verify if tests work on OSX 10.5 | |
| # (10.4 is known to be bad) | |
| raise unittest.SkipTest("Locale support on MacOSX is minimal") | |
| if sys.platform.startswith("win"): | |
| tlocs = ("En", "English") | |
| else: | |
| tlocs = ("en_US.UTF-8", "en_US.US-ASCII", "en_US") | |
| oldlocale = locale.setlocale(locale.LC_NUMERIC) | |
| for tloc in tlocs: | |
| try: | |
| locale.setlocale(locale.LC_NUMERIC, tloc) | |
| except locale.Error: | |
| continue | |
| break | |
| else: | |
| raise unittest.SkipTest( | |
| "Test locale not supported (tried %s)" % (', '.join(tlocs))) | |
| enUS_locale = tloc | |
| locale.setlocale(locale.LC_NUMERIC, oldlocale) | |
| class BaseLocalizedTest(unittest.TestCase): | |
| # | |
| # Base class for tests using a real locale | |
| # | |
| def setUp(self): | |
| self.oldlocale = locale.setlocale(self.locale_type) | |
| locale.setlocale(self.locale_type, enUS_locale) | |
| if verbose: | |
| print "testing with \"%s\"..." % enUS_locale, | |
| def tearDown(self): | |
| locale.setlocale(self.locale_type, self.oldlocale) | |
| class BaseCookedTest(unittest.TestCase): | |
| # | |
| # Base class for tests using cooked localeconv() values | |
| # | |
| def setUp(self): | |
| locale._override_localeconv = self.cooked_values | |
| def tearDown(self): | |
| locale._override_localeconv = {} | |
| class CCookedTest(BaseCookedTest): | |
| # A cooked "C" locale | |
| cooked_values = { | |
| 'currency_symbol': '', | |
| 'decimal_point': '.', | |
| 'frac_digits': 127, | |
| 'grouping': [], | |
| 'int_curr_symbol': '', | |
| 'int_frac_digits': 127, | |
| 'mon_decimal_point': '', | |
| 'mon_grouping': [], | |
| 'mon_thousands_sep': '', | |
| 'n_cs_precedes': 127, | |
| 'n_sep_by_space': 127, | |
| 'n_sign_posn': 127, | |
| 'negative_sign': '', | |
| 'p_cs_precedes': 127, | |
| 'p_sep_by_space': 127, | |
| 'p_sign_posn': 127, | |
| 'positive_sign': '', | |
| 'thousands_sep': '' | |
| } | |
| class EnUSCookedTest(BaseCookedTest): | |
| # A cooked "en_US" locale | |
| cooked_values = { | |
| 'currency_symbol': '$', | |
| 'decimal_point': '.', | |
| 'frac_digits': 2, | |
| 'grouping': [3, 3, 0], | |
| 'int_curr_symbol': 'USD ', | |
| 'int_frac_digits': 2, | |
| 'mon_decimal_point': '.', | |
| 'mon_grouping': [3, 3, 0], | |
| 'mon_thousands_sep': ',', | |
| 'n_cs_precedes': 1, | |
| 'n_sep_by_space': 0, | |
| 'n_sign_posn': 1, | |
| 'negative_sign': '-', | |
| 'p_cs_precedes': 1, | |
| 'p_sep_by_space': 0, | |
| 'p_sign_posn': 1, | |
| 'positive_sign': '', | |
| 'thousands_sep': ',' | |
| } | |
| class FrFRCookedTest(BaseCookedTest): | |
| # A cooked "fr_FR" locale with a space character as decimal separator | |
| # and a non-ASCII currency symbol. | |
| cooked_values = { | |
| 'currency_symbol': '\xe2\x82\xac', | |
| 'decimal_point': ',', | |
| 'frac_digits': 2, | |
| 'grouping': [3, 3, 0], | |
| 'int_curr_symbol': 'EUR ', | |
| 'int_frac_digits': 2, | |
| 'mon_decimal_point': ',', | |
| 'mon_grouping': [3, 3, 0], | |
| 'mon_thousands_sep': ' ', | |
| 'n_cs_precedes': 0, | |
| 'n_sep_by_space': 1, | |
| 'n_sign_posn': 1, | |
| 'negative_sign': '-', | |
| 'p_cs_precedes': 0, | |
| 'p_sep_by_space': 1, | |
| 'p_sign_posn': 1, | |
| 'positive_sign': '', | |
| 'thousands_sep': ' ' | |
| } | |
| class BaseFormattingTest(object): | |
| # | |
| # Utility functions for formatting tests | |
| # | |
| def _test_formatfunc(self, format, value, out, func, **format_opts): | |
| self.assertEqual( | |
| func(format, value, **format_opts), out) | |
| def _test_format(self, format, value, out, **format_opts): | |
| self._test_formatfunc(format, value, out, | |
| func=locale.format, **format_opts) | |
| def _test_format_string(self, format, value, out, **format_opts): | |
| self._test_formatfunc(format, value, out, | |
| func=locale.format_string, **format_opts) | |
| def _test_currency(self, value, out, **format_opts): | |
| self.assertEqual(locale.currency(value, **format_opts), out) | |
| class EnUSNumberFormatting(BaseFormattingTest): | |
| # XXX there is a grouping + padding bug when the thousands separator | |
| # is empty but the grouping array contains values (e.g. Solaris 10) | |
| def setUp(self): | |
| self.sep = locale.localeconv()['thousands_sep'] | |
| def test_grouping(self): | |
| self._test_format("%f", 1024, grouping=1, out='1%s024.000000' % self.sep) | |
| self._test_format("%f", 102, grouping=1, out='102.000000') | |
| self._test_format("%f", -42, grouping=1, out='-42.000000') | |
| self._test_format("%+f", -42, grouping=1, out='-42.000000') | |
| def test_grouping_and_padding(self): | |
| self._test_format("%20.f", -42, grouping=1, out='-42'.rjust(20)) | |
| if self.sep: | |
| self._test_format("%+10.f", -4200, grouping=1, | |
| out=('-4%s200' % self.sep).rjust(10)) | |
| self._test_format("%-10.f", -4200, grouping=1, | |
| out=('-4%s200' % self.sep).ljust(10)) | |
| def test_integer_grouping(self): | |
| self._test_format("%d", 4200, grouping=True, out='4%s200' % self.sep) | |
| self._test_format("%+d", 4200, grouping=True, out='+4%s200' % self.sep) | |
| self._test_format("%+d", -4200, grouping=True, out='-4%s200' % self.sep) | |
| def test_integer_grouping_and_padding(self): | |
| self._test_format("%10d", 4200, grouping=True, | |
| out=('4%s200' % self.sep).rjust(10)) | |
| self._test_format("%-10d", -4200, grouping=True, | |
| out=('-4%s200' % self.sep).ljust(10)) | |
| def test_simple(self): | |
| self._test_format("%f", 1024, grouping=0, out='1024.000000') | |
| self._test_format("%f", 102, grouping=0, out='102.000000') | |
| self._test_format("%f", -42, grouping=0, out='-42.000000') | |
| self._test_format("%+f", -42, grouping=0, out='-42.000000') | |
| def test_padding(self): | |
| self._test_format("%20.f", -42, grouping=0, out='-42'.rjust(20)) | |
| self._test_format("%+10.f", -4200, grouping=0, out='-4200'.rjust(10)) | |
| self._test_format("%-10.f", 4200, grouping=0, out='4200'.ljust(10)) | |
| def test_complex_formatting(self): | |
| # Spaces in formatting string | |
| self._test_format_string("One million is %i", 1000000, grouping=1, | |
| out='One million is 1%s000%s000' % (self.sep, self.sep)) | |
| self._test_format_string("One million is %i", 1000000, grouping=1, | |
| out='One million is 1%s000%s000' % (self.sep, self.sep)) | |
| # Dots in formatting string | |
| self._test_format_string(".%f.", 1000.0, out='.1000.000000.') | |
| # Padding | |
| if self.sep: | |
| self._test_format_string("--> %10.2f", 4200, grouping=1, | |
| out='--> ' + ('4%s200.00' % self.sep).rjust(10)) | |
| # Asterisk formats | |
| self._test_format_string("%10.*f", (2, 1000), grouping=0, | |
| out='1000.00'.rjust(10)) | |
| if self.sep: | |
| self._test_format_string("%*.*f", (10, 2, 1000), grouping=1, | |
| out=('1%s000.00' % self.sep).rjust(10)) | |
| # Test more-in-one | |
| if self.sep: | |
| self._test_format_string("int %i float %.2f str %s", | |
| (1000, 1000.0, 'str'), grouping=1, | |
| out='int 1%s000 float 1%s000.00 str str' % | |
| (self.sep, self.sep)) | |
| class TestFormatPatternArg(unittest.TestCase): | |
| # Test handling of pattern argument of format | |
| def test_onlyOnePattern(self): | |
| # Issue 2522: accept exactly one % pattern, and no extra chars. | |
| self.assertRaises(ValueError, locale.format, "%f\n", 'foo') | |
| self.assertRaises(ValueError, locale.format, "%f\r", 'foo') | |
| self.assertRaises(ValueError, locale.format, "%f\r\n", 'foo') | |
| self.assertRaises(ValueError, locale.format, " %f", 'foo') | |
| self.assertRaises(ValueError, locale.format, "%fg", 'foo') | |
| self.assertRaises(ValueError, locale.format, "%^g", 'foo') | |
| self.assertRaises(ValueError, locale.format, "%f%%", 'foo') | |
| class TestLocaleFormatString(unittest.TestCase): | |
| """General tests on locale.format_string""" | |
| def test_percent_escape(self): | |
| self.assertEqual(locale.format_string('%f%%', 1.0), '%f%%' % 1.0) | |
| self.assertEqual(locale.format_string('%d %f%%d', (1, 1.0)), | |
| '%d %f%%d' % (1, 1.0)) | |
| self.assertEqual(locale.format_string('%(foo)s %%d', {'foo': 'bar'}), | |
| ('%(foo)s %%d' % {'foo': 'bar'})) | |
| def test_mapping(self): | |
| self.assertEqual(locale.format_string('%(foo)s bing.', {'foo': 'bar'}), | |
| ('%(foo)s bing.' % {'foo': 'bar'})) | |
| self.assertEqual(locale.format_string('%(foo)s', {'foo': 'bar'}), | |
| ('%(foo)s' % {'foo': 'bar'})) | |
| class TestNumberFormatting(BaseLocalizedTest, EnUSNumberFormatting): | |
| # Test number formatting with a real English locale. | |
| locale_type = locale.LC_NUMERIC | |
| def setUp(self): | |
| BaseLocalizedTest.setUp(self) | |
| EnUSNumberFormatting.setUp(self) | |
| class TestEnUSNumberFormatting(EnUSCookedTest, EnUSNumberFormatting): | |
| # Test number formatting with a cooked "en_US" locale. | |
| def setUp(self): | |
| EnUSCookedTest.setUp(self) | |
| EnUSNumberFormatting.setUp(self) | |
| def test_currency(self): | |
| self._test_currency(50000, "$50000.00") | |
| self._test_currency(50000, "$50,000.00", grouping=True) | |
| self._test_currency(50000, "USD 50,000.00", | |
| grouping=True, international=True) | |
| class TestCNumberFormatting(CCookedTest, BaseFormattingTest): | |
| # Test number formatting with a cooked "C" locale. | |
| def test_grouping(self): | |
| self._test_format("%.2f", 12345.67, grouping=True, out='12345.67') | |
| def test_grouping_and_padding(self): | |
| self._test_format("%9.2f", 12345.67, grouping=True, out=' 12345.67') | |
| class TestFrFRNumberFormatting(FrFRCookedTest, BaseFormattingTest): | |
| # Test number formatting with a cooked "fr_FR" locale. | |
| def test_decimal_point(self): | |
| self._test_format("%.2f", 12345.67, out='12345,67') | |
| def test_grouping(self): | |
| self._test_format("%.2f", 345.67, grouping=True, out='345,67') | |
| self._test_format("%.2f", 12345.67, grouping=True, out='12 345,67') | |
| def test_grouping_and_padding(self): | |
| self._test_format("%6.2f", 345.67, grouping=True, out='345,67') | |
| self._test_format("%7.2f", 345.67, grouping=True, out=' 345,67') | |
| self._test_format("%8.2f", 12345.67, grouping=True, out='12 345,67') | |
| self._test_format("%9.2f", 12345.67, grouping=True, out='12 345,67') | |
| self._test_format("%10.2f", 12345.67, grouping=True, out=' 12 345,67') | |
| self._test_format("%-6.2f", 345.67, grouping=True, out='345,67') | |
| self._test_format("%-7.2f", 345.67, grouping=True, out='345,67 ') | |
| self._test_format("%-8.2f", 12345.67, grouping=True, out='12 345,67') | |
| self._test_format("%-9.2f", 12345.67, grouping=True, out='12 345,67') | |
| self._test_format("%-10.2f", 12345.67, grouping=True, out='12 345,67 ') | |
| def test_integer_grouping(self): | |
| self._test_format("%d", 200, grouping=True, out='200') | |
| self._test_format("%d", 4200, grouping=True, out='4 200') | |
| def test_integer_grouping_and_padding(self): | |
| self._test_format("%4d", 4200, grouping=True, out='4 200') | |
| self._test_format("%5d", 4200, grouping=True, out='4 200') | |
| self._test_format("%10d", 4200, grouping=True, out='4 200'.rjust(10)) | |
| self._test_format("%-4d", 4200, grouping=True, out='4 200') | |
| self._test_format("%-5d", 4200, grouping=True, out='4 200') | |
| self._test_format("%-10d", 4200, grouping=True, out='4 200'.ljust(10)) | |
| def test_currency(self): | |
| euro = u'\u20ac'.encode('utf-8') | |
| self._test_currency(50000, "50000,00 " + euro) | |
| self._test_currency(50000, "50 000,00 " + euro, grouping=True) | |
| # XXX is the trailing space a bug? | |
| self._test_currency(50000, "50 000,00 EUR ", | |
| grouping=True, international=True) | |
| class TestStringMethods(BaseLocalizedTest): | |
| locale_type = locale.LC_CTYPE | |
| if sys.platform != 'sunos5' and not sys.platform.startswith("win"): | |
| # Test BSD Rune locale's bug for isctype functions. | |
| def test_isspace(self): | |
| self.assertEqual('\x20'.isspace(), True) | |
| self.assertEqual('\xa0'.isspace(), False) | |
| self.assertEqual('\xa1'.isspace(), False) | |
| def test_isalpha(self): | |
| self.assertEqual('\xc0'.isalpha(), False) | |
| def test_isalnum(self): | |
| self.assertEqual('\xc0'.isalnum(), False) | |
| def test_isupper(self): | |
| self.assertEqual('\xc0'.isupper(), False) | |
| def test_islower(self): | |
| self.assertEqual('\xc0'.islower(), False) | |
| def test_lower(self): | |
| self.assertEqual('\xcc\x85'.lower(), '\xcc\x85') | |
| def test_upper(self): | |
| self.assertEqual('\xed\x95\xa0'.upper(), '\xed\x95\xa0') | |
| def test_strip(self): | |
| self.assertEqual('\xed\x95\xa0'.strip(), '\xed\x95\xa0') | |
| def test_split(self): | |
| self.assertEqual('\xec\xa0\xbc'.split(), ['\xec\xa0\xbc']) | |
| class TestMiscellaneous(unittest.TestCase): | |
| def test_getpreferredencoding(self): | |
| # Invoke getpreferredencoding to make sure it does not cause exceptions. | |
| enc = locale.getpreferredencoding() | |
| if enc: | |
| # If encoding non-empty, make sure it is valid | |
| codecs.lookup(enc) | |
| if hasattr(locale, "strcoll"): | |
| def test_strcoll_3303(self): | |
| # test crasher from bug #3303 | |
| self.assertRaises(TypeError, locale.strcoll, u"a", None) | |
| def test_setlocale_category(self): | |
| locale.setlocale(locale.LC_ALL) | |
| locale.setlocale(locale.LC_TIME) | |
| locale.setlocale(locale.LC_CTYPE) | |
| locale.setlocale(locale.LC_COLLATE) | |
| locale.setlocale(locale.LC_MONETARY) | |
| locale.setlocale(locale.LC_NUMERIC) | |
| # crasher from bug #7419 | |
| self.assertRaises(locale.Error, locale.setlocale, 12345) | |
| def test_main(): | |
| tests = [ | |
| TestMiscellaneous, | |
| TestFormatPatternArg, | |
| TestLocaleFormatString, | |
| TestEnUSNumberFormatting, | |
| TestCNumberFormatting, | |
| TestFrFRNumberFormatting, | |
| ] | |
| # SkipTest can't be raised inside unittests, handle it manually instead | |
| try: | |
| get_enUS_locale() | |
| except unittest.SkipTest as e: | |
| if verbose: | |
| print "Some tests will be disabled: %s" % e | |
| else: | |
| tests += [TestNumberFormatting, TestStringMethods] | |
| run_unittest(*tests) | |
| if __name__ == '__main__': | |
| test_main() |