def test_query(self):
conn = MaxComputeConnection(testing.get_datasource())
rs = conn.query("select * from notexist limit 1")
self.assertFalse(rs.success())
self.assertTrue("Table not found" in rs.error())
rs = conn.query(
"select * from alifin_jtest_dev.sqlflow_iris_train limit 1")
self.assertTrue(rs.success())
rows = [r for r in rs]
self.assertEqual(1, len(rows))
rs = conn.query(
"select * from alifin_jtest_dev.sqlflow_iris_train limit 20")
self.assertTrue(rs.success())
col_info = rs.column_info()
self.assertEqual([('sepal_length', 'double'),
('sepal_width', 'double'),
('petal_length', 'double'),
('petal_width', 'double'), ('class', 'bigint')],
col_info)
rows = [r for r in rs]
self.assertTrue(20, len(rows))
def test_query(self):
conn = MaxComputeConnection(testing.get_datasource())
try:
conn.query("select * from notexist limit 1")
self.assertTrue(False)
except Exception as e:
self.assertTrue("Table not found" in str(e))
rs = conn.query(
"select * from alifin_jtest_dev.sqlflow_iris_train limit 1")
self.assertTrue(rs.success())
rows = [r for r in rs]
self.assertEqual(1, len(rows))
rs = conn.query(
"select * from alifin_jtest_dev.sqlflow_iris_train limit 20")
self.assertTrue(rs.success())
col_info = rs.column_info()
self.assertEqual([('sepal_length', 'DOUBLE'),
('sepal_width', 'DOUBLE'),
('petal_length', 'DOUBLE'),
('petal_width', 'DOUBLE'), ('class', 'BIGINT')],
col_info)
rows = [r for r in rs]
self.assertTrue(20, len(rows))
def test_exec(self):
conn = MaxComputeConnection(testing.get_datasource())
rs = conn.exec(
"create table alifin_jtest_dev.sqlflow_test_exec(a int)")
self.assertTrue(rs)
rs = conn.exec(
"insert into alifin_jtest_dev.sqlflow_test_exec values(1), (2)")
self.assertTrue(rs)
rs = conn.query("select * from alifin_jtest_dev.sqlflow_test_exec")
self.assertTrue(rs.success())
rows = [r for r in rs]
self.assertTrue(2, len(rows))
rs = conn.exec("drop table alifin_jtest_dev.sqlflow_test_exec")
self.assertTrue(rs)