Issue #14135: Make sure tests don't leave the locale changed for any

brettcannon · brettcannon · commit 2d556f56db94 · 2013-04-01T12:53:13.000-04:00
category.

Thanks to Westley Martinez for trying to fix the issue and Atsuo
Ishimoto for an initial patch.
diff --git a/Lib/test/regrtest.py b/Lib/test/regrtest.py
@@ -127,6 +127,7 @@
 import faulthandler
 import io
 import json
+import locale
 import logging
 import os
 import platform
@@ -1061,7 +1062,7 @@ def __init__(self, testname, verbose=0, quiet=False):
                  'sys.warnoptions', 'threading._dangling',
                  'multiprocessing.process._dangling',
                  'sysconfig._CONFIG_VARS', 'sysconfig._INSTALL_SCHEMES',
-                 'support.TESTFN',
+                 'support.TESTFN', 'locale'
                 )
 
     def get_sys_argv(self):
@@ -1230,6 +1231,14 @@ def restore_support_TESTFN(self, saved_value):
             elif os.path.isdir(support.TESTFN):
                 shutil.rmtree(support.TESTFN)
 
+    _locale_categories = [getattr(locale, lc)
+                          for lc in dir(locale) if lc.startswith('LC_')]
+    def get_locale(self):
+        return tuple(map(locale.getlocale, self._locale_categories))
+    def restore_locale(self, saved):
+        for category, setting in zip(self._locale_categories, saved):
+            locale.setlocale(category, setting)
+
     def resource_info(self):
         for name in self.resources:
             method_suffix = name.replace('.', '_')
