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)