def test_predict_logic(self): model = BiasedRFClassifier(k_nearest_neighbor=5) model.fit(self.x, self.imbalance_y) pred = model.predict(self.x) self.assertEqual(len(self.imbalance_y), len(pred)) print(pred)
def test_fit_logic(self): model = BiasedRFClassifier(k_nearest_neighbor=5) model.fit(self.x, self.imbalance_y) self.assertTrue(model._is_fitted())