def test_vaca_transpose(self): v = Vaca(config_file=self.config_file, connection_name='default') rows = v.q('''select * from user limit 1;''').transpose() self.assertEqual(len(rows), 2) self.assertEqual(rows[0][0], 'id') self.assertEqual(rows[1][0], 'username') self.assertEqual(rows[0][1], 0) self.assertEqual(rows[1][1], 'batman')
def test_vaca_get_raw(self): v = Vaca(config_file=self.config_file, connection_name='default') rows = v.q('''select * from user limit 1;''').get_raw() self.assertEqual(len(rows), 1) self.assertEqual(rows[0][1], 'batman')
def test_vaca_get(self): v = Vaca(config_file=self.config_file, connection_name='default') rows = v.q('''select * from user limit 1;''').get() self.assertEqual(len(rows), 2) self.assertEqual(rows[0][0], 'id') self.assertEqual(rows[1][0], 0)
def test_vaca_save_as_csv(self): v = Vaca(config_file=self.config_file, connection_name='default') v.q('''select * from user;''').save_as_text('csv.test') self.assertTrue(os.path.isfile('csv.test'))
def test_vaca_show_encoding(self): v = Vaca(config_file=self.config_file, connection_name='default') q = v.q('''select * from user;''').show() self.assertEqual(len(q.query_ret), 5)
def test_vaca_run(self): v = Vaca(config_file=self.config_file, connection_name='default') q = v.q('''select * from user limit 1;''').run() self.assertEqual(len(q.query_ret), 2)
def test_vaca_get_map(self): v = Vaca(config_file=self.config_file, connection_name='default') m = v.q('''select * from user limit 1;''').get_map() m = list(m) self.assertEqual(len(m), 1) self.assertEqual(m[0]['username'], 'batman')
def test_link_from_config(self): Vaca(config_file=self.config_file, connection_name='default') from music.models import Song