-
Notifications
You must be signed in to change notification settings - Fork 2k
/
pyunit_levels_nlevels_setlevel_setLevels.py
99 lines (81 loc) · 4.14 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
import sys
sys.path.insert(1, "../../")
import h2o
def levels_nlevels_setlevel_setLevels_test(ip,port):
# Connect to h2o
h2o.init(ip,port)
iris = h2o.import_frame(path=h2o.locate("smalldata/iris/iris.csv"))
# frame (default)
levels = iris.levels()
nlevels = iris.nlevels()
# frame (w/ index)
levels = iris.levels(col=4)
nlevels = iris.nlevels(col=4)
assert set(['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)
# vec
iris[4] = iris[4].setLevel(level='Iris-setosa')
levels = iris.levels(col=4)
nlevels = iris.nlevels(col=4)
assert set(['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-setosa'
levels = iris[4].levels()
nlevels = iris[4].nlevels()
assert set(['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)
iris[4] = iris[4].setLevel(level='Iris-versicolor')
levels = iris.levels(col=4)
nlevels = iris.nlevels(col=4)
assert set(['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 == None, "Expected levels to be None, 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_frame(path=h2o.locate("smalldata/iris/iris.csv"))
iris[4] = iris[4].setLevels(levels=["a", "b", "c"])
# frame (default)
levels = iris.levels()
nlevels = iris.nlevels()
# frame (w/ index)
levels = iris.levels(col=4)
nlevels = iris.nlevels(col=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].setLevel(level='a')
levels = iris.levels(col=4)
nlevels = iris.nlevels(col=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)
assert iris[0,4] == 'a'
levels = iris[4].levels()
nlevels = iris[4].nlevels()
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].setLevel(level='b')
levels = iris.levels(col=4)
nlevels = iris.nlevels(col=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)
assert iris[0,4] == 'b'
levels = iris[1].levels()
nlevels = iris[1].nlevels()
assert levels == None, "Expected levels to be None, 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.setLevel(level='c')
assert one_column_frame[0,0] == 'c'
if __name__ == "__main__":
h2o.run_test(sys.argv, levels_nlevels_setlevel_setLevels_test)