add tests for the intercept scikit-learn-contrib#214

PascalCarrivain · PascalCarrivain · commit 21b1dee1c9ca · 2024-03-18T16:25:07.000+01:00
diff --git a/skglm/experimental/tests/test_sqrt_lasso.py b/skglm/experimental/tests/test_sqrt_lasso.py
@@ -16,7 +16,7 @@ def test_alpha_max():
 
     sqrt_lasso = SqrtLasso(alpha=alpha_max).fit(X, y)
 
-    np.testing.assert_equal(sqrt_lasso.coef_, 0)
+    np.testing.assert_equal(sqrt_lasso.coef_[:n_features], 0)
 
 
 def test_vs_statsmodels():
