d7dfef48a5afaf3d3a664ca9997baecd50c94cd6,category_encoders/tests/test_count.py,TestCountEncoder,test_count_min_group_size_int,#TestCountEncoder#,29

Before Change

        self.assertEqual(len(set(out["extra"].values)), 4)

        // single group under threshold
        enc = encoders.CountEncoder(verbose=1, min_group_size=30)
        enc.fit(X)
        out = enc.transform(X_t)
        self.assertEqual(len(set(out["extra"].values)), 4)

        // multiple groups under threshold

After Change

        enc.fit(X)
        out = enc.transform(X_t)
        self.assertTrue(np.isin([31, 28, 41], out["none"].unique()).all())
        self.assertTrue(out["none"].unique().shape == (3,))
        self.assertTrue(out["none"].isna().sum() == 0)
        self.assertIn("B_nan", enc.mapping["none"])
        self.assertTrue(np.isin([28, 25, 19], out["na_categorical"].unique()).all())
        self.assertTrue(out["na_categorical"].unique().shape == (3,))

Instances