Exemplos de BubbleScatter._get_encoder_categories em Python

Exemplo n.º 1

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

Exemplo n.º 2

 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"])