def testShapeDict(self):
ref_0_1_2 = {
'a1': { 0.5: { 10: (1,), 15: (2,), }, 0.6: { 10: (3,), 15: (4,), }, },
'a2': { 0.5: { 10: (5,), 15: (6,), }, 0.6: { 10: (7,), 15: (8,), }, }
}
self.assertEquals(shape_dict(self.data0, 0, 1, 2), ref_0_1_2)
ref_1_2_0 = {
0.5: { 10: { 'a1': (1,), 'a2': (5,), }, 15: { 'a1': (2,), 'a2': (6,), }, },
0.6: { 10: { 'a1': (3,), 'a2': (7,), }, 15: { 'a1': (4,), 'a2': (8,), }, },
}
self.assertEquals(shape_dict(self.data0, 1, 2, 0), ref_1_2_0)
ref_2_0_1 = {
10: { 'a1': { 0.5: (1,), 0.6: (3,), }, 'a2': { 0.5: (5,), 0.6: (7,), }, },
15: { 'a1': { 0.5: (2,), 0.6: (4,), }, 'a2': { 0.5: (6,), 0.6: (8,), }, },
}
self.assertEquals(shape_dict(self.data0, 2, 0, 1), ref_2_0_1)
ref_0_1 = {
'a1': { 0.5: (1, 2,), 0.6: (3, 4,), },
'a2': { 0.5: (5, 6,), 0.6: (7, 8,), }
}
self.assertEquals(shape_dict(self.data0, 0, 1), ref_0_1)
ref_0 = {
'a1': (1, 2, 3, 4, ),
'a2': (5, 6, 7, 8, ),
}
self.assertEquals(shape_dict(self.data0, 0), ref_0)
def test_shape_dict():
data = [
(("a1", 0.5, 10, 555), [1]),
(("a1", 0.5, 15, 555), [2]),
(("a1", 0.6, 10, 555), [3]),
(("a1", 0.6, 15, 555), [4]),
(("a2", 0.5, 10, 555), [5]),
(("a2", 0.5, 15, 555), [6]),
(("a2", 0.6, 10, 555), [7]),
(("a2", 0.6, 15, 555), [8]),
(("a1", 0.5, 10, 555), [11]),
(("a1", 0.5, 15, 555), [12]),
(("a1", 0.6, 10, 555), [13]),
(("a1", 0.6, 15, 555), [14]),
(("a2", 0.5, 10, 555), [15]),
(("a2", 0.5, 15, 555), [16]),
(("a2", 0.6, 10, 555), [17]),
(("a2", 0.6, 15, 555), [18]),
]
sd = shape_dict(data, 0, 1, 2)
for k, v in sd.iteritems():
print " ", k
for kk, vv in v.iteritems():
print " ", kk
for kkk, vvv in vv.iteritems():
print " ", kkk, vvv
def plot_multi():
data = [
(("a1", 0.5, 10), np.random.normal(2.0, 0.2, 20)),
(("a1", 0.5, 15), np.random.normal(2.1, 0.2, 20)),
(("a1", 0.6, 10), np.random.normal(2.2, 0.2, 20)),
(("a1", 0.6, 15), np.random.normal(2.3, 0.2, 20)),
(("a2", 0.5, 10), np.random.normal(2.4, 0.2, 20)),
(("a2", 0.5, 15), np.random.normal(2.5, 0.2, 20)),
(("a2", 0.6, 10), np.random.normal(2.6, 0.2, 20)),
(("a2", 0.6, 15), np.random.normal(2.7, 0.2, 20)),
]
sd = shape_dict(data, 2, 1, 0)
v2_order = ["a1", "a2"]
v1_order = [0.5, 0.6]
multi_boxplot(sd)
