def test_font_scale(self):
"""
Test scaling the fonts
"""
notebook_ref = yb_rcmod._plotting_context("notebook")
notebook_big = yb_rcmod._plotting_context("notebook", 2)
font_keys = [
"axes.labelsize", "axes.titlesize", "legend.fontsize",
"xtick.labelsize", "ytick.labelsize", "font.size"
]
for k in font_keys:
self.assertEqual(notebook_ref[k] * 2, notebook_big[k])
def test_context_context_manager(self):
yb_rcmod._set_context("notebook")
orig_params = yb_rcmod._plotting_context()
context_params = yb_rcmod._plotting_context("paper")
with yb_rcmod._plotting_context("paper"):
self.assert_rc_params(context_params)
self.assert_rc_params(orig_params)
@yb_rcmod._plotting_context("paper")
def func():
self.assert_rc_params(context_params)
func()
self.assert_rc_params(orig_params)
def test__set_context(self):
"""
Test setting the context
"""
context_dict = yb_rcmod._plotting_context()
yb_rcmod._set_context()
self.assert_rc_params(context_dict)
def test_rc_override(self):
"""
Test overriding the rc params dictionary
"""
key, val = "grid.linewidth", 5
rc = {key: val, "foo": "bar"}
out = yb_rcmod._plotting_context("talk", rc=rc)
assert out[key] == val
assert "foo" not in out
def test_rc_override(self):
"""
Test overriding the rc params dictionary
"""
key, val = "grid.linewidth", 5
rc = {key: val, "foo": "bar"}
out = yb_rcmod._plotting_context("talk", rc=rc)
self.assertEqual(out[key], val)
self.assertNotIn("foo", out)
def test_default_return(self):
"""
Test the context returns the default
"""
current = yb_rcmod._plotting_context()
self.assert_rc_params(current)