def test_style_context_manager(self): rcmod.set_style("darkgrid") orig_params = rcmod.axes_style() with rcmod.axes_style("whitegrid"): context_params = rcmod.axes_style("whitegrid") self.assert_rc_params(context_params) self.assert_rc_params(orig_params)
def test_set_style(self): for style in self.styles: style_dict = rcmod.axes_style(style) rcmod.set_style(style) self.assert_rc_params(style_dict)
def test_rc_override(self): rc = {"axes.facecolor": "blue", "foo.notaparam": "bar"} out = rcmod.axes_style("darkgrid", rc) self.assertEqual(out["axes.facecolor"], "blue") self.assertNotIn("foo.notaparam", out)
def test_bad_style(self): with self.assertRaises(ValueError): rcmod.axes_style("i_am_not_a_style")
def test_key_usage(self): _style_keys = set(rcmod._style_keys) for style in self.styles: nt.assert_true(not set(rcmod.axes_style(style)) ^ _style_keys)
def test_default_return(self): current = rcmod.axes_style() self.assert_rc_params(current)