Merge pull request #913 from Kaggle/add-xgboost-version-regression-test

Add xgboost version regression test.
Kaggle · Dec 8, 2020 · f1a3bb3 · f1a3bb3
2 parents cb38e56 + d60116a
commit f1a3bb3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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