Skip to content

Commit f300279

Browse files
TST: test new read_sql api
1 parent e6a531d commit f300279

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

pandas/io/tests/test_sql.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -566,6 +566,19 @@ def test_read_table_index_col(self):
566566
self.assertEqual(result.columns.tolist(), ["C", "D"],
567567
"columns not set correctly whith index_col")
568568

569+
def test_read_sql_delegate(self):
570+
iris_frame1 = sql.read_sql_query(
571+
"SELECT * FROM iris", self.conn)
572+
iris_frame2 = sql.read_sql(
573+
"SELECT * FROM iris", self.conn)
574+
tm.assert_frame_equal(iris_frame1, iris_frame2,
575+
"read_sql and read_sql_query have not the same"
576+
" result with a query")
577+
578+
iris_frame1 = sql.read_sql_table('iris', self.conn)
579+
iris_frame2 = sql.read_sql('iris', self.conn)
580+
tm.assert_frame_equal(iris_frame1, iris_frame2)
581+
569582

570583
class TestSQLLegacyApi(_TestSQLApi):
571584

@@ -620,6 +633,18 @@ def test_sql_open_close(self):
620633

621634
tm.assert_frame_equal(self.test_frame2, result)
622635

636+
def test_read_sql_delegate(self):
637+
iris_frame1 = sql.read_sql_query(
638+
"SELECT * FROM iris", self.conn, flavor=self.flavor)
639+
iris_frame2 = sql.read_sql(
640+
"SELECT * FROM iris", self.conn, flavor=self.flavor)
641+
tm.assert_frame_equal(iris_frame1, iris_frame2,
642+
"read_sql and read_sql_query have not the same"
643+
" result with a query")
644+
645+
self.assertRaises(ValueError, sql.read_sql, 'iris', self.conn,
646+
flavor=self.flavor)
647+
623648

624649
class _TestSQLAlchemy(PandasSQLTest):
625650
"""

0 commit comments

Comments
 (0)