bugfix for predict with gamma = 'auto'

1 files changed, 17 insertions, 0 deletions

diff --git a/test/test_core.py b/test/test_core.py
index 0a1042a..63c172d 100644
--- a/test/test_core.py
+++ b/test/test_core.py
@@ -197,6 +197,23 @@ class GenSVMTestCase(unittest.TestCase):
             set(pred).issubset(set(["versicolor", "virginica", "setosa"]))
         )
 
+    def test_fit_nonlinear_auto(self):
+        """ GENSVM: Fit and predict with nonlinear kernel """
+        data = load_iris()
+        X = data.data
+        y = data.target_names[data.target]
+
+        X_train, X_test, y_train, y_test = train_test_split(
+            X, y, random_state=123
+        )
+        clf = GenSVM(kernel="rbf", max_iter=1000, random_state=123)
+        clf.fit(X_train, y_train)
+
+        pred = clf.predict(X_test, trainX=X_train)
+        self.assertTrue(
+            set(pred).issubset(set(["versicolor", "virginica", "setosa"]))
+        )
+
     def test_fit_with_seed(self):
         """ GENSVM: Test fit with seeding """