bugfix and STRidge test (#14)

Author: TarenGorman

sparsereg/model/base.py

@@ -148,7 +148,7 @@ def fit(self, x_, y, sample_weight=None):
         if self.threshold > 0:
             self._reduce(x, y)
         else:
-            self.coef_ = self._regress(x[:, ind], y, self.alpha)
+            self.coef_ = self._regress(x[:, self.ind_], y, self.alpha)
 
         if self.unbias and self.alpha >= 0:
             self._unbias(x, y)

test/model/test_stridge.py

@@ -36,3 +36,9 @@ def test_all_zero(data):
     s = STRidge(threshold=10000).fit(x, y)
     assert not any(s.coef_)
     assert len(s.history_) == 1 # initial guess wipes everything
+
+def test_all_nonzero(data):
+    x, y = data
+    s = STRidge(threshold=0).fit(x, y)
+    assert s.complexity == 3
+    assert len(s.history_) == 1 # initial guess is final guess