def test_input_set_generator(self):
"""Test that we're generating data sets to input"""
input = """#int\tstr\tfloat
1\tasd\t0.0
2\t123\t0.2
3\t\t0.3
4\tzxc\t0.4
5\tqaz\t0.5""".splitlines()
con = MockConnection()
cursor = con.cursor()
types = ['i','s','f']
gen = input_set_generator(input,cursor,types,2)
exp1 = [[1,2],['asd','123'],[0.0,0.2]]
exp2 = [[3,4],['','zxc'],[0.3,]]
exp3 = [[5],['qaz'],[0.5]]
obs1 = gen.next()
obs2 = gen.next()
obs3 = gen.next()
self.assertRaises(StopIteration, gen.next)
self.assertEqual(obs1, exp1)
self.assertEqual(obs2, exp2)
self.assertEqual(obs3, exp3)
def test_unzip_and_cast_to_cxoracle_types(self):
"""Tests that we can unzip and cast"""
input = [['1','asd','0.0'],
['2','123','0.2'],
['3','qwe','0.3'],
['4','zxc','0.4'],
['5','qaz','0.5']]
con = MockConnection()
cursor = con.cursor()
types = ['i','s','f']
obs = unzip_and_cast_to_cxoracle_types(input,cursor,types)
exp = [[1,2,3,4,5],['asd','123','qwe','zxc','qaz'],
[0.0,0.2,0.3,0.4,0.5]]
self.assertEqual(obs, exp)
def test_cursor(self):
mc = MockConnection()
con = mc.connect()
cur = con.cursor()
self.assertSameObj(mc, cur)
def test_arrayvar(self):
mc = MockConnection()
obs = mc.arrayvar(int, ['1','2','3'])
exp = [1,2,3]
self.assertEqual(obs, exp)
def test_connect(self):
mc = MockConnection()
con = mc.connect(user='******',password='******',dsn='asd')
self.assertSameObj(mc, con)
