def test_fixed_init_rf(self): model = BaseModel('c-rf', 'test_params_rf_set') model.fit(X, y) self.assertEqual(model.predict_proba(X).shape[0], n_samples) self.assertEqual(model.predict_proba(X).shape[1], num_classes)