summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorAntoine Pitrou <solipsis@pitrou.net>2008-07-25 21:45:08 (GMT)
committerAntoine Pitrou <solipsis@pitrou.net>2008-07-25 21:45:08 (GMT)
commit83d6a87a40711c6012ef856f4a2e6e7b2c30cedf (patch)
tree955f041d0f827f5660c0823d9eee0480b9448750 /Lib
parent6e1df8d0d44817a7b9810c7d8b5c81423fb52cdb (diff)
downloadcpython-83d6a87a40711c6012ef856f4a2e6e7b2c30cedf.zip
cpython-83d6a87a40711c6012ef856f4a2e6e7b2c30cedf.tar.gz
cpython-83d6a87a40711c6012ef856f4a2e6e7b2c30cedf.tar.bz2
Merged revisions 65237 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk ........ r65237 | antoine.pitrou | 2008-07-25 22:40:19 +0200 (ven., 25 juil. 2008) | 3 lines convert test_locale to unittest, and add a mechanism to override localconv() results for further testing (#1864, #1222) ........
Diffstat (limited to 'Lib')
-rw-r--r--Lib/locale.py16
-rw-r--r--Lib/test/test_locale.py308
2 files changed, 240 insertions, 84 deletions
diff --git a/Lib/locale.py b/Lib/locale.py
index 2f8cd85..0bed19e 100644
--- a/Lib/locale.py
+++ b/Lib/locale.py
@@ -13,6 +13,7 @@
import sys, encodings, encodings.aliases
from builtins import str as _builtin_str
+import functools
# Try importing the _locale module.
#
@@ -94,6 +95,21 @@ if 'strxfrm' not in globals():
if 'strcoll' not in globals():
strcoll = _strcoll
+
+_localeconv = localeconv
+
+# With this dict, you can override some items of localeconv's return value.
+# This is useful for testing purposes.
+_override_localeconv = {}
+
+@functools.wraps(_localeconv)
+def localeconv():
+ d = _localeconv()
+ if _override_localeconv:
+ d.update(_override_localeconv)
+ return d
+
+
### Number formatting APIs
# Author: Martin von Loewis
diff --git a/Lib/test/test_locale.py b/Lib/test/test_locale.py
index b1ff14a..251dc7a 100644
--- a/Lib/test/test_locale.py
+++ b/Lib/test/test_locale.py
@@ -1,89 +1,229 @@
-from test.support import verbose, TestSkipped, TestFailed
+from test.support import run_unittest, TestSkipped, verbose
+import unittest
import locale
import sys
+import codecs
-if sys.platform == 'darwin':
- raise TestSkipped(
- "Locale support on MacOSX is minimal and cannot be tested")
-oldlocale = locale.setlocale(locale.LC_NUMERIC)
-
-if sys.platform.startswith("win"):
- tlocs = ("En", "English")
-else:
- tlocs = ("en_US.UTF-8", "en_US.US-ASCII", "en_US")
-
-for tloc in tlocs:
- try:
- locale.setlocale(locale.LC_NUMERIC, tloc)
- break
- except locale.Error:
- continue
-else:
- raise ImportError(
- "test locale not supported (tried %s)" % (', '.join(tlocs)))
-
-def testformat(formatstr, value, grouping = 0, output=None, func=locale.format):
- if verbose:
- if output:
- print("%s %% %s =? %s ..." %
- (repr(formatstr), repr(value), repr(output)), end=' ')
- else:
- print("%s %% %s works? ..." % (repr(formatstr), repr(value)),
- end=' ')
- result = func(formatstr, value, grouping = grouping)
- if output and result != output:
- if verbose:
- print('no')
- print("%s %% %s == %s != %s" %
- (repr(formatstr), repr(value), repr(result), repr(output)))
+class BaseLocalizedTest(unittest.TestCase):
+ #
+ # Base class for tests using a real locale
+ #
+
+ if sys.platform.startswith("win"):
+ tlocs = ("En", "English")
else:
+ tlocs = ("en_US.UTF-8", "en_US.US-ASCII", "en_US")
+
+ def setUp(self):
+ if sys.platform == 'darwin':
+ raise TestSkipped(
+ "Locale support on MacOSX is minimal and cannot be tested")
+ self.oldlocale = locale.setlocale(self.locale_type)
+ for tloc in self.tlocs:
+ try:
+ locale.setlocale(self.locale_type, tloc)
+ except locale.Error:
+ continue
+ break
+ else:
+ raise TestSkipped(
+ "Test locale not supported (tried %s)" % (', '.join(self.tlocs)))
if verbose:
- print("yes")
-
-try:
- # On Solaris 10, the thousands_sep is the empty string
- sep = locale.localeconv()['thousands_sep']
- testformat("%f", 1024, grouping=1, output='1%s024.000000' % sep)
- testformat("%f", 102, grouping=1, output='102.000000')
- testformat("%f", -42, grouping=1, output='-42.000000')
- testformat("%+f", -42, grouping=1, output='-42.000000')
- testformat("%20.f", -42, grouping=1, output=' -42')
- testformat("%+10.f", -4200, grouping=1, output=' -4%s200' % sep)
- testformat("%-10.f", 4200, grouping=1, output='4%s200 ' % sep)
- # Invoke getpreferredencoding to make sure it does not cause exceptions,
- locale.getpreferredencoding()
-
- # === Test format() with more complex formatting strings
- # test if grouping is independent from other characters in formatting string
- testformat("One million is %i", 1000000, grouping=1,
- output='One million is 1%s000%s000' % (sep, sep),
- func=locale.format_string)
- testformat("One million is %i", 1000000, grouping=1,
- output='One million is 1%s000%s000' % (sep, sep),
- func=locale.format_string)
- # test dots in formatting string
- testformat(".%f.", 1000.0, output='.1000.000000.', func=locale.format_string)
- # test floats
- testformat("--> %10.2f", 1000.0, grouping=1, output='--> 1%s000.00' % sep,
- func=locale.format_string)
- # test asterisk formats
- testformat("%10.*f", (2, 1000.0), grouping=0, output=' 1000.00',
- func=locale.format_string)
- testformat("%*.*f", (10, 2, 1000.0), grouping=1, output=' 1%s000.00' % sep,
- func=locale.format_string)
- # test more-in-one
- testformat("int %i float %.2f str %s", (1000, 1000.0, 'str'), grouping=1,
- output='int 1%s000 float 1%s000.00 str str' % (sep, sep),
- func=locale.format_string)
-
-finally:
- locale.setlocale(locale.LC_NUMERIC, oldlocale)
-
-if hasattr(locale, "strcoll"):
- # test crasher from bug #3303
- try:
- locale.strcoll("a", None)
- except TypeError:
- pass
- else:
- raise TestFailed("TypeError not raised")
+ print("testing with \"%s\"..." % tloc, 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 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):
+
+ def setUp(self):
+ # NOTE: On Solaris 10, the thousands_sep is the empty string
+ 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))
+ 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_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
+ 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))
+ self._test_format_string("%*.*f", (10, 2, 1000), grouping=1,
+ out=('1%s000.00' % self.sep).rjust(10))
+ # Test more-in-one
+ 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 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 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, "a", None)
+ self.assertRaises(TypeError, locale.strcoll, b"a", None)
+
+
+def test_main():
+ run_unittest(__name__)
+
+if __name__ == '__main__':
+ test_main()