Skip to content

Commit f1a3bb3

Browse files
authored
Merge pull request #913 from Kaggle/add-xgboost-version-regression-test
Add xgboost version regression test.
2 parents cb38e56 + d60116a commit f1a3bb3

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/test_xgboost.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,16 @@
11
import unittest
22

3+
import xgboost
4+
5+
from distutils.version import StrictVersion
36
from sklearn import datasets
47
from xgboost import XGBClassifier
58

69
class TestXGBoost(unittest.TestCase):
10+
def test_version(self):
11+
# b/175051617 prevent xgboost version downgrade.
12+
self.assertGreaterEqual(StrictVersion(xgboost.__version__), StrictVersion("1.2.1"))
13+
714
def test_classifier(self):
815
boston = datasets.load_boston()
916
X, y = boston.data, boston.target

0 commit comments

Comments
 (0)