diff --git a/tests/test_xgboost.py b/tests/test_xgboost.py index e1bc12f8..00e1beb4 100644 --- a/tests/test_xgboost.py +++ b/tests/test_xgboost.py @@ -1,9 +1,16 @@ import unittest +import xgboost + +from distutils.version import StrictVersion from sklearn import datasets from xgboost import XGBClassifier class TestXGBoost(unittest.TestCase): + def test_version(self): + # b/175051617 prevent xgboost version downgrade. + self.assertGreaterEqual(StrictVersion(xgboost.__version__), StrictVersion("1.2.1")) + def test_classifier(self): boston = datasets.load_boston() X, y = boston.data, boston.target