BUG: incorrect model_file results in segfault

2015-09-16 21:53:51 +09:00
parent ae43fd7c7a
commit db0c9e1c2d
2 changed files with 15 additions and 1 deletions
--- a/tests/python/test_basic.py
+++ b/tests/python/test_basic.py
@@ -1,9 +1,20 @@
 # -*- coding: utf-8 -*-
 import numpy as np
 import xgboost as xgb
+import unittest
+

 dpath = 'demo/data/'

+
+class TestBasic(unittest.TestCase):
+
+    def test_load_file_invalid(self):
+
+        self.assertRaises(ValueError, xgb.Booster,
+                          model_file='incorrect_path')
+
+
 def test_basic():
    dtrain = xgb.DMatrix(dpath + 'agaricus.txt.train')
    dtest = xgb.DMatrix(dpath + 'agaricus.txt.test')