def test_font_scale(self):
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):
for context in self.contexts:
context_dict = yb_rcmod.plotting_context(context)
yb_rcmod.set_context(context)
self.assert_rc_params(context_dict)
def test_rc_override(self):
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_bad_context(self):
with self.assertRaises(ValueError):
yb_rcmod.plotting_context("i_am_not_a_context")
def test_key_usage(self):
_context_keys = set(yb_rcmod._context_keys)
for context in self.contexts:
missing = set(yb_rcmod.plotting_context(context)) ^ _context_keys
self.assertTrue(not missing)
def test_default_return(self):
current = yb_rcmod.plotting_context()
self.assert_rc_params(current)