From d60116aecbb6935fae508c94905a335fdb0603bb Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Mon, 7 Dec 2020 23:55:57 +0000 Subject: [PATCH] Add xgboost version regression test. BUG=175051617 --- tests/test_xgboost.py | 7 +++++++ 1 file changed, 7 insertions(+) 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