def test_plot_show_true(self): df = pd.DataFrame() df["x"] = [1, 1, 2, 2] df["y"] = [1, 2, 1, 1] scatter = BubbleScatter(df, "x", "y") result = scatter.plot(show=True) self.assertIsNone(result)
def test_plot_show_false(self): df = pd.DataFrame() df["x"] = [1, 1, 2, 2] df["y"] = [1, 2, 1, 1] scatter = BubbleScatter(df, "x", "y") result = scatter.plot(show=False) self.assertIsInstance(result, plt.Figure)