TST: Fixed rcparams unicode parsing test

Kojoley · Kojoley · commit e5cb3a6eaaf5 · 2016-10-22T01:40:18.000+03:00
diff --git a/lib/matplotlib/tests/test_rcparams.py b/lib/matplotlib/tests/test_rcparams.py
@@ -10,6 +10,10 @@
 
 from cycler import cycler, Cycler
 
+try:
+    from unittest import mock
+except ImportError:
+    import mock
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 from matplotlib.tests import assert_str_equal
@@ -224,16 +228,13 @@ def test_legend_edgecolor():
         yield _legend_rcparam_helper, rc_dict, target, get_func
 
 
-def test_Issue_1713():
+def test_unicode_parsing():
+    # Related issue #1713
     utf32_be = os.path.join(os.path.dirname(__file__),
                            'test_utf32_be_rcparams.rc')
-    old_lang = os.environ.get('LANG', None)
-    os.environ['LANG'] = 'en_US.UTF-32-BE'
-    rc = mpl.rc_params_from_file(utf32_be, True)
-    if old_lang:
-        os.environ['LANG'] = old_lang
-    else:
-        del os.environ['LANG']
+    import locale
+    with mock.patch('locale.getpreferredencoding', return_value='UTF-32-BE'):
+        rc = mpl.rc_params_from_file(utf32_be, True, False)
     assert rc.get('timezone') == 'UTC'
 
 
