added valueError check.

dswah · Jul 15, 2020 · f411c35 · f411c35
1 parent bfd9488
commit f411c35
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pygam/tests/test_GAM_methods.py b/pygam/tests/test_GAM_methods.py
@@ -260,6 +260,10 @@ def test_predict_terms_output(mcycle_X_y):
     n,m = x.shape
     assert(len(terms) == m)
     assert(len(terms[0]) == n)
+    try:
+        terms = gam.predict(x,output = "shyambhu")
+    except:
+        assert(True)
 
 class TestSamplingFromPosterior(object):