TST improve coverage of fastica function (#15102)

NicolasHug · rth · commit a07392f7899c · 2019-09-27T17:31:20.000+02:00
diff --git a/sklearn/decomposition/tests/test_fastica.py b/sklearn/decomposition/tests/test_fastica.py
@@ -280,3 +280,22 @@ def test_fastica_errors():
     with pytest.raises(ValueError, match='Invalid algorithm.+must '
                        'be.+parallel.+or.+deflation'):
         fastica(X, algorithm='pizza')
+
+
+@pytest.mark.parametrize('whiten', [True, False])
+@pytest.mark.parametrize('return_X_mean', [True, False])
+@pytest.mark.parametrize('return_n_iter', [True, False])
+def test_fastica_output_shape(whiten, return_X_mean, return_n_iter):
+    n_features = 3
+    n_samples = 10
+    rng = np.random.RandomState(0)
+    X = rng.random_sample((n_samples, n_features))
+
+    expected_len = 3 + return_X_mean + return_n_iter
+
+    out = fastica(X, whiten=whiten, return_n_iter=return_n_iter,
+                  return_X_mean=return_X_mean)
+
+    assert len(out) == expected_len
+    if not whiten:
+        assert out[0] is None
