def test4(self): df = DataFrame() df[1] = range(100) df[2] = ['bob' for i in range(100)] df[3] = [i * 1.234232 for i in range(100)] df[4] = ['bob' for i in range(50)] + range(50) with self.assertRaises(TypeError) as cm: df._build_sqlite3_tbl(df.keys()[:2], 42) self.assertEqual(str(cm.exception), "'int' object is not iterable")
def test31(self): df = DataFrame() df[1] = range(100) df[2] = ['bob' for i in range(100)] df[3] = [i * 1.234232 for i in range(100)] df[4] = ['bob' for i in range(50)] + range(50) shuffle(df[1]) shuffle(df[2]) shuffle(df[3]) with self.assertRaises(KeyError) as cm: df._build_sqlite3_tbl(df.keys()[:2], ['4 != "bob"']) self.assertEqual(str(cm.exception), "'4'")
def test3(self): """test with string keys and tuple where condition""" df = DataFrame() df[1] = range(100) df[2] = ['bob' for i in range(100)] df[3] = [i * 1.234232 for i in range(100)] df[4] = ['bob' for i in range(50)] + range(50) shuffle(df[1]) shuffle(df[2]) shuffle(df[3]) df._build_sqlite3_tbl(df.keys()[:2], [(4, '!=', 'bob')]) df._execute('select * from TBL') for i, (a, b) in enumerate(df.cur): self.assertEqual(a, df[1][i + 50]) self.assertEqual(b, df[2][i + 50])
def test22(self): """test with string keys and where condition""" df = DataFrame() df['1'] = range(100) df['2'] = ['bob' for i in range(100)] df['3'] = [i * 1.234232 for i in range(100)] df['4'] = ['bob' for i in range(50)] + range(50) shuffle(df['1']) shuffle(df['2']) shuffle(df['3']) df._build_sqlite3_tbl(df.keys()[:2], ['4 not in ("bob")']) df._execute('select * from TBL') for i, (a, b) in enumerate(df.cur): self.assertEqual(a, df['1'][i + 50]) self.assertEqual(b, df['2'][i + 50])
def test1(self): """test with integer keys subset of table""" df = DataFrame() df[1] = range(100) df[2] = ['bob' for i in range(100)] df[3] = [i * 1.234232 for i in range(100)] df[4] = ['bob' for i in range(50)] + range(50) shuffle(df[1]) shuffle(df[2]) shuffle(df[3]) shuffle(df[4]) df._build_sqlite3_tbl(df.keys()[:2]) df._execute('select * from TBL') for i, (a, b) in enumerate(df.cur): self.assertEqual(a, df[1][i]) self.assertEqual(b, df[2][i])
def test02(self): """test with tuple keys""" df = DataFrame() ## df.PRINTQUERIES = True df[(1, )] = range(100) df[(2, )] = ['bob' for i in range(100)] df[(3, )] = [i * 1.234232 for i in range(100)] df[(4, )] = ['bob' for i in range(50)] + range(50) shuffle(df[(1, )]) shuffle(df[(2, )]) shuffle(df[(3, )]) shuffle(df[(4, )]) df._build_sqlite3_tbl(df.keys()) df._execute('select * from TBL') for i, (a, b, c, d) in enumerate(df.cur): self.assertEqual(a, df[(1, )][i]) self.assertEqual(b, df[(2, )][i]) self.assertEqual(c, df[(3, )][i]) self.assertEqual(d, str(df[(4, )][i]))
def test00(self): """test with string keys""" df = DataFrame() ## df.PRINTQUERIES=True df['1'] = range(100) df['2'] = ['bob' for i in range(100)] df['3'] = [i * 1.234232 for i in range(100)] df['4'] = ['bob' for i in range(50)] + range(50) df['5'] = np.sqrt(df['3'] * 100.) ## print(df) shuffle(df['1']) shuffle(df['2']) shuffle(df['3']) df._build_sqlite3_tbl(df.keys()) df._execute('select * from TBL') for i, (a, b, c, d, e) in enumerate(df.cur): self.assertEqual(a, df['1'][i]) self.assertEqual(b, df['2'][i]) self.assertEqual(c, df['3'][i]) self.assertEqual(d, str(df['4'][i]))