wip4

scikit-learn · Jun 2, 2018 · 401c65f · 401c65f
1 parent d916d39
commit 401c65f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/sklearn/decomposition/tests/test_dict_learning.py b/sklearn/decomposition/tests/test_dict_learning.py
@@ -312,9 +312,11 @@ def test_sparse_encode_positivity(positive):
         else:
             assert_true((code < 0).any())
 
-    assert_raises(
-        ValueError, sparse_encode, X, V, algorithm='omp', positive=positive
-    )
+    try:
+        sparse_encode(X, V, algorithm='omp', positive=positive)
+    except ValueError:
+        if not positive:
+            raise
 
 
 def test_sparse_encode_input():