Add SQL read_parameters tests

Author: gpoulin

pandas/io/tests/test_sql.py

@@ -95,6 +95,24 @@
                 INSERT INTO types_test_data
                 VALUES(%s, %s, %s, %s, %s, %s, %s, %s)
                 """
+    },
+    'read_parameters': {
+        'sqlite': "SELECT * FROM iris WHERE Name=? AND SepalLength=?",
+        'mysql': 'SELECT * FROM iris WHERE `Name`="%s" AND `SepalLength`=%s',
+        'postgresql': 'SELECT * FROM iris WHERE "Name"=%s AND "SepalLength"=%s'
+    },
+    'read_named_parameters': {
+        'sqlite': """
+                SELECT * FROM iris WHERE Name=:name AND SepalLength=:length
+                """,
+        'mysql': """
+                SELECT * FROM iris WHERE
+                `Name`="%(name)s" AND `SepalLength`=%(length)s
+                """,
+        'postgresql': """
+                SELECT * FROM iris WHERE
+                "Name"=%(name)s AND "SepalLength"=%(length)s
+                """
     }
 }
 
@@ -168,6 +186,18 @@ def _read_sql_iris(self):
         iris_frame = self.pandasSQL.read_sql("SELECT * FROM iris")
         self._check_iris_loaded_frame(iris_frame)
 
+    def _read_sql_iris_parameter(self):
+        query = SQL_STRINGS['read_parameters'][self.flavor]
+        params = ['Iris-setosa', 5.1]
+        iris_frame = self.pandasSQL.read_sql(query, params=params)
+        self._check_iris_loaded_frame(iris_frame)
+
+    def _read_sql_iris_named_parameter(self):
+        query = SQL_STRINGS['read_named_parameters'][self.flavor]
+        params = {'name': 'Iris-setosa', 'length': 5.1}
+        iris_frame = self.pandasSQL.read_sql(query, params=params)
+        self._check_iris_loaded_frame(iris_frame)
+
     def _to_sql(self):
         self.drop_table('test_frame1')
 
@@ -491,6 +521,12 @@ class _TestSQLAlchemy(PandasSQLTest):
     def test_read_sql(self):
         self._read_sql_iris()
 
+    def test_read_sql_parameter(self):
+        self._read_sql_iris_parameter()
+
+    def test_read_sql_named_parameter(self):
+        self._read_sql_iris_named_parameter()
+
     def test_to_sql(self):
         self._to_sql()
 
@@ -703,6 +739,12 @@ def test_invalid_flavor(self):
     def test_read_sql(self):
         self._read_sql_iris()
 
+    def test_read_sql_parameter(self):
+        self._read_sql_iris_parameter()
+
+    def test_read_sql_named_parameter(self):
+        self._read_sql_iris_named_parameter()
+
     def test_to_sql(self):
         self._to_sql()