def test_setup_with_seaborn_off(self): """Test setup configures seaborn disabled """ seaborn = mock.MagicMock() conf['pyexperiment.plot.seaborn.enable'] = False with mock.patch.dict('sys.modules', {'seaborn': seaborn}): plot.setup_plotting() self.assertEqual([], seaborn.method_calls) del conf['pyexperiment']
def test_setup_without_seaborn(self): """Test setup mentions missing seaborn """ with mock.patch.dict('sys.modules', {'seaborn': None}): log_stream = StringIO() Logger.CONSOLE_STREAM_HANDLER = logging.StreamHandler(log_stream) log.reset_instance() log.initialize() plot.setup_plotting() self.assertRegexpMatches(log_stream.getvalue(), r'Cannot import seaborn')
def test_setup_with_seaborn(self): """Test setup configures seaborn """ seaborn = mock.MagicMock() options = {'seaborn.style': 'foo', 'seaborn.palette_name': 'bar', 'seaborn.desat': 12} with mock.patch.dict('sys.modules', {'seaborn': seaborn}): plot.setup_plotting(options=options) self.assertIn(mock.call.set_style('foo'), seaborn.method_calls) self.assertIn(mock.call.set_palette('bar', desat=12), seaborn.method_calls)
def test_setup_with_seaborn(self): """Test setup configures seaborn """ seaborn = mock.MagicMock() options = { 'seaborn.style': 'foo', 'seaborn.palette_name': 'bar', 'seaborn.desat': 12 } with mock.patch.dict('sys.modules', {'seaborn': seaborn}): plot.setup_plotting(options=options) self.assertIn(mock.call.set_style('foo'), seaborn.method_calls) self.assertIn(mock.call.set_palette('bar', desat=12), seaborn.method_calls)
def test_setup_fig_setup_plotting(self): """Test setting up a figure sets up plotting as well """ _fig = plot.setup_figure() self.assertFalse(plot.setup_plotting(override_setup=False))
def test_setup_does_override(self): """Test setting up plotting with override """ self.assertTrue(plot.setup_plotting(override_setup=True)) self.assertTrue(plot.setup_plotting(override_setup=True))
def test_setup_does_not_override(self): """Test setting up plotting """ self.assertTrue(plot.setup_plotting(override_setup=False)) self.assertFalse(plot.setup_plotting(override_setup=False))