Add xgboost version regression test.

BUG=175051617
Kaggle · Dec 7, 2020 · d60116a · d60116a
1 parent cb38e56
commit d60116a
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