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 test4(self):
df = DataFrame()
df[1] = list(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)] + list(range(50))
with self.assertRaises(TypeError) as cm:
df._build_sqlite3_tbl(list(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 test31(self):
df = DataFrame()
df[1] = list(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)] + list(range(50))
shuffle(df[1])
shuffle(df[2])
shuffle(df[3])
with self.assertRaises(KeyError) as cm:
df._build_sqlite3_tbl(list(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 test3(self):
"""test with string keys and tuple where condition"""
df = DataFrame()
df[1] = list(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)] + list(range(50))
shuffle(df[1])
shuffle(df[2])
shuffle(df[3])
df._build_sqlite3_tbl(list(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'] = list(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)] + list(range(50))
shuffle(df['1'])
shuffle(df['2'])
shuffle(df['3'])
df._build_sqlite3_tbl(list(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 test1(self):
"""test with integer keys subset of table"""
df = DataFrame()
df[1] = list(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)] + list(range(50))
shuffle(df[1])
shuffle(df[2])
shuffle(df[3])
shuffle(df[4])
df._build_sqlite3_tbl(list(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 test01(self):
"""test with integer keys"""
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())
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 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 test02(self):
"""test with tuple keys"""
df = DataFrame()
## df.PRINTQUERIES = True
df[(1, )] = list(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)] + list(range(50))
shuffle(df[(1, )])
shuffle(df[(2, )])
shuffle(df[(3, )])
shuffle(df[(4, )])
df._build_sqlite3_tbl(list(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.0)
## 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]))
def test00(self):
"""test with string keys"""
df = DataFrame()
## df.PRINTQUERIES=True
df['1'] = list(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)] + list(range(50))
df['5'] = np.sqrt(df['3'] * 100.)
## print(df)
shuffle(df['1'])
shuffle(df['2'])
shuffle(df['3'])
df._build_sqlite3_tbl(list(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]))
