Skip to content

Commit 9c4f034

Browse files
Update sql unit tests
1 parent bcbe5ea commit 9c4f034

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

pandas/io/sql.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1715,7 +1715,13 @@ def execute(self, *args, **kwargs):
17151715

17161716
@staticmethod
17171717
def _query_iterator(
1718-
cursor, chunksize, columns, index_col=None, coerce_float=True, parse_dates=None
1718+
cursor,
1719+
chunksize,
1720+
columns,
1721+
index_col=None,
1722+
coerce_float=True,
1723+
parse_dates=None,
1724+
dtype=None,
17191725
):
17201726
"""Return generator through chunked result set"""
17211727
while True:
@@ -1732,6 +1738,7 @@ def _query_iterator(
17321738
index_col=index_col,
17331739
coerce_float=coerce_float,
17341740
parse_dates=parse_dates,
1741+
dtype=dtype,
17351742
)
17361743

17371744
def read_query(
@@ -1742,6 +1749,7 @@ def read_query(
17421749
params=None,
17431750
parse_dates=None,
17441751
chunksize=None,
1752+
dtype=None,
17451753
):
17461754

17471755
args = _convert_params(sql, params)
@@ -1756,6 +1764,7 @@ def read_query(
17561764
index_col=index_col,
17571765
coerce_float=coerce_float,
17581766
parse_dates=parse_dates,
1767+
dtype=dtype,
17591768
)
17601769
else:
17611770
data = self._fetchall_as_list(cursor)
@@ -1767,6 +1776,7 @@ def read_query(
17671776
index_col=index_col,
17681777
coerce_float=coerce_float,
17691778
parse_dates=parse_dates,
1779+
dtype=dtype,
17701780
)
17711781
return frame
17721782

0 commit comments

Comments
 (0)