def test_setup_fig_setup_figsize(self): """Test setting up a figure sets up figure with size """ with mock.patch('pyexperiment.utils.plot.plt') as plt: _fig = plot.setup_figure(figsize=(2, 3)) self.assertIn(mock.call.figure(figsize=(2, 3)), plt.mock_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_figure(self): """Test setting up a figure """ fig = plot.setup_figure() self.assertIsNotNone(fig)