def test__get_encoder_categories_with_bins_no_categories(self): df = pd.DataFrame(columns=["x", "y"]) scatter = BubbleScatter(df, "x", "y", xbins=3, ybins=3) x_categories = scatter._get_encoder_categories("x") y_categories = scatter._get_encoder_categories("y") self.assertEqual(x_categories, "auto") self.assertEqual(y_categories, "auto")
def test__get_encoder_categories_no_bins_with_categories(self): df = pd.DataFrame(columns=["x", "y"]) scatter = BubbleScatter(df, "x", "y", x_categories=["a", "b", "c"], y_categories=["q", "w", "e"]) x_categories = scatter._get_encoder_categories("x") y_categories = scatter._get_encoder_categories("y") self.assertListEqual(x_categories, ["a", "b", "c"]) self.assertListEqual(y_categories, ["q", "w", "e"])