[TEST] Minor smoke test for missing data

netneurolab · Nov 4, 2019 · 4fbd403 · 4fbd403
1 parent 6bd5ab8
commit 4fbd403
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/pyls/tests/types/test_regression.py b/pyls/tests/types/test_regression.py
@@ -80,6 +80,17 @@ def test_regression_3dbootstrap(aggfunc):
                        bootsamples=bootsamples, n_boot=10)
 
 
+def test_regression_missingdata():
+    X = rs.rand(subj, Xf)
+    X[10] = np.nan
+    PLSRegressionTests(X=X, n_components=2)
+    X[20] = np.nan
+    PLSRegressionTests(X=X, n_components=2)
+    Y = rs.rand(subj, Yf)
+    Y[11] = np.nan
+    PLSRegressionTests(X=X, Y=Y, n_components=2)
+
+
 def test_errors():
     with pytest.raises(ValueError):
         PLSRegressionTests(n_components=1000)