mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 10:26:02 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			439 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			439 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| from 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().release.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")
 | |
|     elif sys.platform.startswith("win"):
 | |
|         tlocs = ("En", "English")
 | |
|     else:
 | |
|         tlocs = ("en_US.UTF-8", "en_US.ISO8859-1", "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, end=' ')
 | |
| 
 | |
|     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': '\u20ac',
 | |
|         '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 = '\u20ac'
 | |
|         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 TestCollation(unittest.TestCase):
 | |
|     # Test string collation functions
 | |
| 
 | |
|     def test_strcoll(self):
 | |
|         self.assertLess(locale.strcoll('a', 'b'), 0)
 | |
|         self.assertEqual(locale.strcoll('a', 'a'), 0)
 | |
|         self.assertGreater(locale.strcoll('b', 'a'), 0)
 | |
| 
 | |
|     def test_strxfrm(self):
 | |
|         self.assertLess(locale.strxfrm('a'), locale.strxfrm('b'))
 | |
| 
 | |
| 
 | |
| class TestEnUSCollation(BaseLocalizedTest, TestCollation):
 | |
|     # Test string collation functions with a real English locale
 | |
| 
 | |
|     locale_type = locale.LC_ALL
 | |
| 
 | |
|     def setUp(self):
 | |
|         enc = codecs.lookup(locale.getpreferredencoding(False) or 'ascii').name
 | |
|         if enc not in ('utf-8', 'iso8859-1', 'cp1252'):
 | |
|             raise unittest.SkipTest('encoding not suitable')
 | |
|         if enc != 'iso8859-1' and (sys.platform == 'darwin' or
 | |
|                                    sys.platform.startswith('freebsd')):
 | |
|             raise unittest.SkipTest('wcscoll/wcsxfrm have known bugs')
 | |
|         BaseLocalizedTest.setUp(self)
 | |
| 
 | |
|     def test_strcoll_with_diacritic(self):
 | |
|         self.assertLess(locale.strcoll('à', 'b'), 0)
 | |
| 
 | |
|     def test_strxfrm_with_diacritic(self):
 | |
|         self.assertLess(locale.strxfrm('à'), locale.strxfrm('b'))
 | |
| 
 | |
| 
 | |
| 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)
 | |
| 
 | |
|     def test_strcoll_3303(self):
 | |
|         # test crasher from bug #3303
 | |
|         self.assertRaises(TypeError, locale.strcoll, "a", None)
 | |
|         self.assertRaises(TypeError, locale.strcoll, b"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_getsetlocale_issue1813(self):
 | |
|         # Issue #1813: setting and getting the locale under a Turkish locale
 | |
|         oldlocale = locale.setlocale(locale.LC_CTYPE)
 | |
|         self.addCleanup(locale.setlocale, locale.LC_CTYPE, oldlocale)
 | |
|         try:
 | |
|             locale.setlocale(locale.LC_CTYPE, 'tr_TR')
 | |
|         except locale.Error:
 | |
|             # Unsupported locale on this system
 | |
|             self.skipTest('test needs Turkish locale')
 | |
|         loc = locale.getlocale(locale.LC_CTYPE)
 | |
|         if verbose:
 | |
|             print('got locale %a' % (loc,))
 | |
|         locale.setlocale(locale.LC_CTYPE, loc)
 | |
|         self.assertEqual(loc, locale.getlocale(locale.LC_CTYPE))
 | |
| 
 | |
|     def test_invalid_locale_format_in_localetuple(self):
 | |
|         with self.assertRaises(TypeError):
 | |
|             locale.setlocale(locale.LC_ALL, b'fi_FI')
 | |
| 
 | |
|     def test_invalid_iterable_in_localetuple(self):
 | |
|         with self.assertRaises(TypeError):
 | |
|             locale.setlocale(locale.LC_ALL, (b'not', b'valid'))
 | |
| 
 | |
| 
 | |
| def test_main():
 | |
|     tests = [
 | |
|         TestMiscellaneous,
 | |
|         TestFormatPatternArg,
 | |
|         TestLocaleFormatString,
 | |
|         TestEnUSNumberFormatting,
 | |
|         TestCNumberFormatting,
 | |
|         TestFrFRNumberFormatting,
 | |
|         TestCollation
 | |
|     ]
 | |
|     # 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, TestEnUSCollation]
 | |
|     run_unittest(*tests)
 | |
| 
 | |
| if __name__ == '__main__':
 | |
|     test_main()
 | 
