-
Notifications
You must be signed in to change notification settings - Fork 2k
/
pyunit_levels_nlevels_setlevel_setLevels.py
105 lines (85 loc) · 4.21 KB
/
pyunit_levels_nlevels_setlevel_setLevels.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import sys
sys.path.insert(1,"../../")
import h2o
from tests import pyunit_utils
def levels_nlevels_setlevel_setLevels_test():
iris = h2o.import_file(path=pyunit_utils.locate("smalldata/iris/iris.csv"))
# frame (default)
levels = iris.levels()
nlevels = iris.nlevels()
# frame (w/ index)
levels = iris[4].levels()[0]
nlevels = iris[4].nlevels()[0]
assert {'Iris-setosa', 'Iris-versicolor', 'Iris-virginica'} == set(levels), \
"Expected levels to be {0}, but got {1}".format(
{'Iris-setosa', 'Iris-versicolor', 'Iris-virginica'},levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
# vec
iris[4] = iris[4].set_level(level='Iris-setosa')
levels = iris[4].levels()[0]
nlevels = iris[4].nlevels()[0]
assert {'Iris-setosa', 'Iris-versicolor', 'Iris-virginica'} == set(levels), \
"Expected levels to be {0}, but got {1}".format(
{'Iris-setosa', 'Iris-versicolor', 'Iris-virginica'},levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
assert iris[0,4] == 'Iris-setosa'
levels = iris[4].levels()[0]
nlevels = iris[4].nlevels()[0]
assert {'Iris-setosa', 'Iris-versicolor', 'Iris-virginica'} == set(levels), \
"Expected levels to be {0}, but got {1}".format(
{'Iris-setosa', 'Iris-versicolor', 'Iris-virginica'},levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
iris[4] = iris[4].set_level(level='Iris-versicolor')
levels = iris[4].levels()[0]
nlevels = iris.nlevels()[4]
assert {'Iris-setosa', 'Iris-versicolor', 'Iris-virginica'} == set(levels), \
"Expected levels to be {0}, but got {1}".format(set(['Iris-setosa', 'Iris-versicolor', 'Iris-virginica']),levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
assert iris[0,4] == 'Iris-versicolor'
levels = iris[1].levels()
nlevels = iris[1].nlevels()
assert levels == [], "Expected levels to be [], but got {0}".format(levels)
assert nlevels == 0, "Expected nlevels to be 0, but got {0}".format(nlevels)
################### reimport, set new domains, rerun tests ###################################
iris = h2o.import_file(path=pyunit_utils.locate("smalldata/iris/iris.csv"))
iris[4] = iris[4].set_levels(levels=["a", "b", "c"])
# frame (default)
levels = iris.levels()
nlevels = iris.nlevels()
# frame (w/ index)
levels = iris[4].levels()[0]
nlevels = iris.nlevels()[4]
assert set(['a', 'b', 'c']) == set(levels), \
"Expected levels to be {0}, but got {1}".format(set(['a', 'b', 'c']),levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
# vec
iris[4] = iris[4].set_level(level='a')
levels = iris[4].levels()[0]
nlevels = iris[4].nlevels()[0]
assert set(['a', 'b', 'c']) == set(levels), \
"Expected levels to be {0}, but got {1}".format(set(['a', 'b', 'c']),levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
assert iris[0,4] == 'a'
levels = iris[4].levels()[0]
nlevels = iris[4].nlevels()[0]
assert set(['a', 'b', 'c']) == set(levels), \
"Expected levels to be {0}, but got {1}".format(set(['a', 'b', 'c']),levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
iris[4] = iris[4].set_level(level='b')
levels = iris[4].levels()[0]
nlevels = iris[4].nlevels()[0]
assert set(['a', 'b', 'c']) == set(levels), \
"Expected levels to be {0}, but got {1}".format(set(['a', 'b', 'c']),levels)
assert nlevels == 3, "Expected nlevels to be 3, but got {0}".format(nlevels)
assert iris[0,4] == 'b'
levels = iris[1].levels()
nlevels = iris[1].nlevels()
assert levels == [], "Expected levels to be [], but got {0}".format(levels)
assert nlevels == 0, "Expected nlevels to be 0, but got {0}".format(nlevels)
one_column_frame = iris[4]
one_column_frame = one_column_frame.set_level(level='c')
assert one_column_frame[0,0] == 'c'
if __name__ == "__main__":
pyunit_utils.standalone_test(levels_nlevels_setlevel_setLevels_test)
else:
levels_nlevels_setlevel_setLevels_test()