Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changed 4 tests

Browse Source 
Changed symbol test to give error on < sign, not on = sign
Changed 3 other functions, so that float is used instead of q
This commit is contained in:
Johan Manders 2015-11-03 21:32:47 +01:00
parent f9e1b2b7b7
commit b0f38e9352
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/python/test_basic.py
View File

@ -48,7 +48,7 @@ class TestBasic(unittest.TestCase):
feature_names=['a', 'b', 'c', 'd', 'd']) feature_names=['a', 'b', 'c', 'd', 'd'])
# contains symbol # contains symbol
self.assertRaises(ValueError, xgb.DMatrix, data, self.assertRaises(ValueError, xgb.DMatrix, data,
feature_names=['a', 'b', 'c', 'd', 'e=1']) feature_names=['a', 'b', 'c', 'd', 'e<1'])
dm = xgb.DMatrix(data) dm = xgb.DMatrix(data)
dm.feature_names = list('abcde') dm.feature_names = list('abcde')
@ -105,7 +105,7 @@ class TestBasic(unittest.TestCase):
df = pd.DataFrame([[1, 2., True], [2, 3., False]], columns=['a', 'b', 'c']) df = pd.DataFrame([[1, 2., True], [2, 3., False]], columns=['a', 'b', 'c'])
dm = xgb.DMatrix(df, label=pd.Series([1, 2])) dm = xgb.DMatrix(df, label=pd.Series([1, 2]))
assert dm.feature_names == ['a', 'b', 'c'] assert dm.feature_names == ['a', 'b', 'c']
assert dm.feature_types == ['int', 'q', 'i'] assert dm.feature_types == ['int', 'float', 'i']
assert dm.num_row() == 2 assert dm.num_row() == 2
assert dm.num_col() == 3 assert dm.num_col() == 3
@ -125,14 +125,14 @@ class TestBasic(unittest.TestCase):
df = pd.DataFrame([[1, 2., True], [2, 3., False]]) df = pd.DataFrame([[1, 2., True], [2, 3., False]])
dm = xgb.DMatrix(df, label=pd.Series([1, 2])) dm = xgb.DMatrix(df, label=pd.Series([1, 2]))
assert dm.feature_names == ['0', '1', '2'] assert dm.feature_names == ['0', '1', '2']
assert dm.feature_types == ['int', 'q', 'i'] assert dm.feature_types == ['int', 'float', 'i']
assert dm.num_row() == 2 assert dm.num_row() == 2
assert dm.num_col() == 3 assert dm.num_col() == 3
df = pd.DataFrame([[1, 2., 1], [2, 3., 1]], columns=[4, 5, 6]) df = pd.DataFrame([[1, 2., 1], [2, 3., 1]], columns=[4, 5, 6])
dm = xgb.DMatrix(df, label=pd.Series([1, 2])) dm = xgb.DMatrix(df, label=pd.Series([1, 2]))
assert dm.feature_names == ['4', '5', '6'] assert dm.feature_names == ['4', '5', '6']
assert dm.feature_types == ['int', 'q', 'int'] assert dm.feature_types == ['int', 'float', 'int']
assert dm.num_row() == 2 assert dm.num_row() == 2
assert dm.num_col() == 3 assert dm.num_col() == 3
@ -293,4 +293,4 @@ class TestBasic(unittest.TestCase):
assert isinstance(g, Digraph) assert isinstance(g, Digraph)
ax = xgb.plot_tree(classifier, num_trees=0) ax = xgb.plot_tree(classifier, num_trees=0)
assert isinstance(ax, Axes) assert isinstance(ax, Axes)