Skip to content

Commit 72248f0

Browse files
committed
TST make test more stable.
1 parent ac2d29d commit 72248f0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/tests/test_cross_validation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def test_kfold_valueerrors():
101101
assert_equal(w[0].category, Warning)
102102
# checking it's the right warning. This might be a bad test since it's
103103
# a characteristic of the code and not a behavior
104-
assert_equal(w[0].lineno, 368)
104+
assert_true("The least populated class" in str(w[0]))
105105

106106
# Error when number of folds is <= 0
107107
assert_raises(ValueError, cval.KFold, 2, 0)

0 commit comments

Comments
 (0)