Skip to content

Commit 4dfbca1

Browse files
ahojnnesogrisel
authored andcommitted
Add test for invalid value of min_samples
1 parent af30cf8 commit 4dfbca1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

sklearn/linear_model/tests/test_ransac.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -210,6 +210,9 @@ def test_ransac_min_n_samples():
210210
residual_threshold=5, random_state=0)
211211
ransac_estimator6 = RANSACRegressor(base_estimator,
212212
residual_threshold=5, random_state=0)
213+
ransac_estimator7 = RANSACRegressor(base_estimator,
214+
min_samples=X.shape[0] + 1,
215+
residual_threshold=5, random_state=0)
213216

214217
ransac_estimator1.fit(X, y)
215218
ransac_estimator2.fit(X, y)
@@ -221,6 +224,7 @@ def test_ransac_min_n_samples():
221224
assert_equal(ransac_estimator1.predict(X), ransac_estimator6.predict(X))
222225
assert_raises(ValueError, ransac_estimator3.fit, X, y)
223226
assert_raises(ValueError, ransac_estimator4.fit, X, y)
227+
assert_raises(ValueError, ransac_estimator7.fit, X, y)
224228

225229

226230
def test_ransac_multi_dimensional_targets():

0 commit comments

Comments
 (0)