Skip to content

Commit e523283

Browse files
committed
Revert "Don't special case for Engine"
This reverts commit 2dfe32c.
1 parent 2dfe32c commit e523283

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

pandas/tests/io/test_sql.py

+12-4
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,7 @@ def create_and_load_postgres_datetz(conn):
251251
Table,
252252
insert,
253253
)
254+
from sqlalchemy.engine import Engine
254255

255256
metadata = MetaData()
256257
datetz = Table("datetz", metadata, Column("DateColWithTz", DateTime(timezone=True)))
@@ -263,10 +264,17 @@ def create_and_load_postgres_datetz(conn):
263264
},
264265
]
265266
stmt = insert(datetz).values(datetz_data)
266-
with conn.begin():
267-
datetz.drop(conn, checkfirst=True)
268-
datetz.create(bind=conn)
269-
conn.execute(stmt)
267+
if isinstance(conn, Engine):
268+
with conn.connect() as conn:
269+
with conn.begin():
270+
datetz.drop(conn, checkfirst=True)
271+
datetz.create(bind=conn)
272+
conn.execute(stmt)
273+
else:
274+
with conn.begin():
275+
datetz.drop(conn, checkfirst=True)
276+
datetz.create(bind=conn)
277+
conn.execute(stmt)
270278

271279
# "2000-01-01 00:00:00-08:00" should convert to
272280
# "2000-01-01 08:00:00"

0 commit comments

Comments
 (0)