def test__transform_with_int_ybins(self):
df = pd.DataFrame()
df["x"] = [0, 1, 2, 3, 4]
df["y"] = [0, 1, 2, 3, 4]
scatter = BubbleScatter(df, "x", "y", ybins=2)
result = scatter._transform("y")
true_result = pd.Series([1.0, 1.0, 1.0, 3.0, 3.0], name="y")
self.assertTrue(result.equals(true_result))
def test__transform_no_ybins(self):
df = pd.DataFrame()
df["x"] = [1, 2, 3]
df["y"] = [1, 2, 3]
scatter = BubbleScatter(df, "x", "y")
result = scatter._transform("y")
true_result = df["x"]
self.assertTrue(result.equals(true_result))