From 4efec8aa479be05cd548f91458ac26a4ecff34f2 Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Wed, 6 Mar 2019 19:15:18 -0500
Subject: Bugfix and test for predict method of GridSearch

---
 test/test_gridsearch.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'test')

diff --git a/test/test_gridsearch.py b/test/test_gridsearch.py
index 1a29b0a..f4cff7e 100644
--- a/test/test_gridsearch.py
+++ b/test/test_gridsearch.py
@@ -172,6 +172,9 @@ class GenSVMGridSearchCVTestCase(unittest.TestCase):
 
         self.assertTrue(hasattr(clf, "best_params_"))
 
+        y_pred = clf.predict(X_test, trainX=X_train)
+        del y_pred
+
     def test_invalid_y(self):
         """ GENSVM_GRID: Check raises for invalid y type """
         pg = {"lmd": [1e-4, 100, 10000], "kernel": ["rbf"]}
-- 
cgit v1.2.3