update test to include predict_proba

Author: mabryj2

test/test_automl/test_estimators.py

@@ -103,7 +103,6 @@ def test_fit_pSMAC(self):
                                        initial_configurations_via_metalearning=0,
                                        ensemble_size=0)
         automl.fit(X_train, Y_train)
-
         # Create a 'dummy model' for the first run, which has an accuracy of
         # more than 99%; it should be in the final ensemble if the ensemble
         # building of the second AutoSklearn classifier works correct
@@ -303,6 +302,8 @@ def test_multilabel(self):
         self.assertEqual(predictions.shape, (50, 3))
         score = f1_macro(Y_test, predictions)
         self.assertGreaterEqual(score, 0.9)
+        probs = automl.predict_proba(X_train)
+        self.assertAlmostEqual(np.mean(probs), 0.33333333333333331)
 
     def test_binary(self):
         output = os.path.join(self.test_dir, '..', '.tmp_binary_fit')
@@ -376,4 +377,4 @@ def test_conversion_of_list_to_np(self, fit_ensemble, refit, fit):
         self.assertIsInstance(refit.call_args[0][1], np.ndarray)
         automl.fit_ensemble(y)
         self.assertEqual(fit_ensemble.call_count, 1)
-        self.assertIsInstance(fit_ensemble.call_args[0][0], np.ndarray)
+        self.assertIsInstance(fit_ensemble.call_args[0][0], np.ndarray)