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)