fix incredibly weird nnz behavior

scikit-learn-contrib · Jun 10, 2017 · 178d48a · 178d48a
1 parent f06f871
commit 178d48a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lightning/impl/tests/test_primal_cd.py b/lightning/impl/tests/test_primal_cd.py
@@ -178,11 +178,11 @@ def test_debiasing_l1l2():
                            multiclass=False,
                            debiasing=True,
                            warm_debiasing=warm_debiasing,
-                           max_iter=20, C=0.01, random_state=0,
-                           verbose=True)
+                           max_iter=20, C=0.01, random_state=0)
         clf.fit(mult_csc, mult_target)
         assert_greater(clf.score(mult_csc, mult_target), 0.75)
-        assert_equal(clf.n_nonzero(percentage=True), 0.08)
+        nnz = clf.n_nonzero(percentage=True)
+        assert_equal(nnz, 0.08)
 
 
 def test_debiasing_warm_start():