Add test_rcparams.py to test rc_file and rc_context.

memmett · memmett · commit 45cba0fb7fc1 · 2012-08-19T12:15:11.000-04:00
diff --git a/lib/matplotlib/tests/mpl.rc b/lib/matplotlib/tests/mpl.rc
@@ -0,0 +1 @@
+lines.linewidth: 33
diff --git a/lib/matplotlib/tests/test_rcparams.py b/lib/matplotlib/tests/test_rcparams.py
@@ -0,0 +1,35 @@
+import os
+
+import matplotlib as mpl
+mpl.rc('text', usetex=False)
+mpl.rc('lines', linewidth=22)
+
+fname = os.path.abspath(os.path.dirname(__file__)) + os.sep + 'mpl.rc'
+
+def test_rcparams():
+
+    usetex = mpl.rcParams['text.usetex']
+    linewidth = mpl.rcParams['lines.linewidth']
+
+    # test context given dictionary
+    with mpl.rc_context(rc={'text.usetex': not usetex}):
+        assert mpl.rcParams['text.usetex'] == (not usetex)
+    assert mpl.rcParams['text.usetex'] == usetex
+
+    # test context given filename (mpl.rc sets linewdith to 33)
+    with mpl.rc_context(fname=fname):
+        assert mpl.rcParams['lines.linewidth'] == 33
+    assert mpl.rcParams['lines.linewidth'] == linewidth
+
+    # test context given filename and dictionary
+    with mpl.rc_context(fname=fname, rc={'lines.linewidth': 44}):
+        assert mpl.rcParams['lines.linewidth'] == 44
+    assert mpl.rcParams['lines.linewidth'] == linewidth
+
+    # test rc_file
+    mpl.rc_file(fname)
+    assert mpl.rcParams['lines.linewidth'] == 33
+    
+
+if __name__ == '__main__':
+    test_rcparams()
