def test3(self): df = DataFrame() with self.assertRaises(Exception) as cm: df.box_plot('a', output_dir='output') self.assertEqual(str(cm.exception), 'Table must have data to print data')
def test6(self): df = DataFrame() df['a'] = [2, 5] df['b'] = [2, 3] with self.assertRaises(KeyError) as cm: df.box_plot('c', output_dir='output') self.assertEqual(str(cm.exception), "'c'")
def test5(self): df = DataFrame() df['a'] = [2, 5] df['b'] = [2, 3] with self.assertRaises(Exception) as cm: df.box_plot('a', 42, output_dir='output') self.assertEqual(str(cm.exception), "'int' object is not iterable")
def test4(self): df = DataFrame() df['a'] = [2] df['b'] = [2, 3] with self.assertRaises(Exception) as cm: df.box_plot('a', output_dir='output') self.assertEqual(str(cm.exception), 'columns have unequal lengths')
def test1(self): R = { 'd': [ np.array([ 9, 8, 6, 8, 10, 4, 6, 5, 7, 7, 7, 9, 6, 6, 6, 11, 6, 3, 8, 7, 11, 13, 8, 6, 14, 11, 13, 13, 10, 11, 12, 11, 16, 11, 9, 23, 12, 10, 19, 11, 10, 19, 14, 5, 10, 11, 14, 15, 11, 11 ]), np.array([ 8, 6, 4, 6, 7, 6, 5, 7, 9, 7, 10, 7, 8, 10, 4, 7, 10, 6, 7, 7, 14, 11, 18, 14, 13, 22, 17, 16, 12, 11, 20, 16, 16, 15, 18, 16, 20, 22, 14, 19, 21, 19, 17, 15, 22, 16, 22, 22, 18, 21 ]) ], 'fname': r'output/box(WORDS~AGE).png', 'maintitle': 'WORDS by AGE', 'xlabels': [u'AGE = old', u'AGE = young'] } df = DataFrame() df.TESTMODE = True df.read_tbl(r'data/words~ageXcondition.csv') D = df.box_plot('WORDS', ['AGE'], output_dir='output') self.assertEqual(D['fname'], R['fname']) self.assertEqual(D['maintitle'], R['maintitle']) self.assertEqual(D['xlabels'], R['xlabels']) for d, r in zip(np.array(D['d']).flat, np.array(R['d']).flat): self.assertAlmostEqual(d, r)
def test0(self): R = { 'd': [ 9.0, 8.0, 6.0, 8.0, 10.0, 4.0, 6.0, 5.0, 7.0, 7.0, 7.0, 9.0, 6.0, 6.0, 6.0, 11.0, 6.0, 3.0, 8.0, 7.0, 11.0, 13.0, 8.0, 6.0, 14.0, 11.0, 13.0, 13.0, 10.0, 11.0, 12.0, 11.0, 16.0, 11.0, 9.0, 23.0, 12.0, 10.0, 19.0, 11.0, 10.0, 19.0, 14.0, 5.0, 10.0, 11.0, 14.0, 15.0, 11.0, 11.0, 8.0, 6.0, 4.0, 6.0, 7.0, 6.0, 5.0, 7.0, 9.0, 7.0, 10.0, 7.0, 8.0, 10.0, 4.0, 7.0, 10.0, 6.0, 7.0, 7.0, 14.0, 11.0, 18.0, 14.0, 13.0, 22.0, 17.0, 16.0, 12.0, 11.0, 20.0, 16.0, 16.0, 15.0, 18.0, 16.0, 20.0, 22.0, 14.0, 19.0, 21.0, 19.0, 17.0, 15.0, 22.0, 16.0, 22.0, 22.0, 18.0, 21.0 ], 'fname': r'output/box(WORDS).png', 'maintitle': 'WORDS', 'val': 'WORDS' } df = DataFrame() df.TESTMODE = True df.read_tbl(r'data/words~ageXcondition.csv') D = df.box_plot('WORDS', output_dir='output') self.assertEqual(D['fname'], R['fname']) self.assertEqual(D['maintitle'], R['maintitle']) self.assertEqual(D['val'], R['val']) for d, r in zip(np.array(D['d']).flat, np.array(R['d']).flat): self.assertAlmostEqual(d, r)
def test2(self): R = { 'd': [ np.array([11, 13, 8, 6, 14, 11, 13, 13, 10, 11]), np.array([9, 8, 6, 8, 10, 4, 6, 5, 7, 7]), np.array([12, 11, 16, 11, 9, 23, 12, 10, 19, 11]), np.array([10, 19, 14, 5, 10, 11, 14, 15, 11, 11]), np.array([7, 9, 6, 6, 6, 11, 6, 3, 8, 7]), np.array([14, 11, 18, 14, 13, 22, 17, 16, 12, 11]), np.array([8, 6, 4, 6, 7, 6, 5, 7, 9, 7]), np.array([20, 16, 16, 15, 18, 16, 20, 22, 14, 19]), np.array([21, 19, 17, 15, 22, 16, 22, 22, 18, 21]), np.array([10, 7, 8, 10, 4, 7, 10, 6, 7, 7]) ], 'fname': r'output/box(WORDS~AGE_X_CONDITION).png', 'maintitle': 'WORDS by AGE * CONDITION', 'xlabels': [ u'AGE = old\nCONDITION = adjective', u'AGE = old\nCONDITION = counting', u'AGE = old\nCONDITION = imagery', u'AGE = old\nCONDITION = intention', u'AGE = old\nCONDITION = rhyming', u'AGE = young\nCONDITION = adjective', u'AGE = young\nCONDITION = counting', u'AGE = young\nCONDITION = imagery', u'AGE = young\nCONDITION = intention', u'AGE = young\nCONDITION = rhyming' ] } df = DataFrame() df.TESTMODE = True df.read_tbl(r'data/words~ageXcondition.csv') D = df.box_plot('WORDS', ['AGE', 'CONDITION'], output_dir='output') self.assertEqual(D['fname'], R['fname']) self.assertEqual(D['maintitle'], R['maintitle']) self.assertEqual(D['xlabels'], R['xlabels']) for d, r in zip(np.array(D['d']).flat, np.array(R['d']).flat): self.assertAlmostEqual(d, r)