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_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_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))
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))
