Fix testcase after update and allow hdfs load

2016-02-26 17:04:51 -08:00
parent be810e4f16
commit 758a77de9c
1 changed files with 2 additions and 2 deletions
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -184,10 +184,10 @@ class TestBasic(unittest.TestCase):

    def test_load_file_invalid(self):

-        self.assertRaises(ValueError, xgb.Booster,
+        self.assertRaises(xgb.XGBoostError, xgb.Booster,
                          model_file='incorrect_path')

-        self.assertRaises(ValueError, xgb.Booster,
+        self.assertRaises(xgb.XGBoostError, xgb.Booster,
                          model_file=u'不正なパス')

    def test_dmatrix_numpy_init(self):