Skip to content

Commit ecb6c4d

Browse files
committed
make test_sql.py mysql_connectable tests parallelizable
1 parent cc35414 commit ecb6c4d

File tree

1 file changed

+11
-9
lines changed

1 file changed

+11
-9
lines changed

pandas/tests/io/test_sql.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1257,6 +1257,7 @@ def test_default_type_conversion(conn, request):
12571257
@pytest.mark.parametrize("conn", mysql_connectable)
12581258
def test_read_procedure(conn, request):
12591259
conn = request.getfixturevalue(conn)
1260+
table_uuid = table_uuid_gen("test_frame")
12601261

12611262
# GH 7324
12621263
# Although it is more an api test, it is added to the
@@ -1265,14 +1266,14 @@ def test_read_procedure(conn, request):
12651266
from sqlalchemy.engine import Engine
12661267

12671268
df = DataFrame({"a": [1, 2, 3], "b": [0.1, 0.2, 0.3]})
1268-
df.to_sql(name="test_frame", con=conn, index=False)
1269+
df.to_sql(name=table_uuid, con=conn, index=False)
12691270

1270-
proc = """DROP PROCEDURE IF EXISTS get_testdb;
1271+
proc = f"""DROP PROCEDURE IF EXISTS get_testdb;
12711272
12721273
CREATE PROCEDURE get_testdb ()
12731274
12741275
BEGIN
1275-
SELECT * FROM test_frame;
1276+
SELECT * FROM {table_uuid};
12761277
END"""
12771278
proc = text(proc)
12781279
if isinstance(conn, Engine):
@@ -1419,6 +1420,7 @@ def test_to_sql_on_public_schema(conn, request):
14191420
def test_insertion_method_on_conflict_update(conn, request):
14201421
# GH 14553: Example in to_sql docstring
14211422
conn = request.getfixturevalue(conn)
1423+
table_uuid = table_uuid_gen("test_insert_conflict")
14221424

14231425
from sqlalchemy.dialects.mysql import insert
14241426
from sqlalchemy.engine import Engine
@@ -1432,8 +1434,8 @@ def insert_on_conflict(table, conn, keys, data_iter):
14321434
return result.rowcount
14331435

14341436
create_sql = text(
1435-
"""
1436-
CREATE TABLE test_insert_conflict (
1437+
f"""
1438+
CREATE TABLE {table_uuid} (
14371439
a INT PRIMARY KEY,
14381440
b FLOAT,
14391441
c VARCHAR(10)
@@ -1449,23 +1451,23 @@ def insert_on_conflict(table, conn, keys, data_iter):
14491451
conn.execute(create_sql)
14501452

14511453
df = DataFrame([[1, 2.1, "a"]], columns=list("abc"))
1452-
df.to_sql(name="test_insert_conflict", con=conn, if_exists="append", index=False)
1454+
df.to_sql(name=table_uuid, con=conn, if_exists="append", index=False)
14531455

14541456
expected = DataFrame([[1, 3.2, "b"]], columns=list("abc"))
14551457
inserted = expected.to_sql(
1456-
name="test_insert_conflict",
1458+
name=table_uuid,
14571459
con=conn,
14581460
index=False,
14591461
if_exists="append",
14601462
method=insert_on_conflict,
14611463
)
1462-
result = sql.read_sql_table("test_insert_conflict", conn)
1464+
result = sql.read_sql_table(table_uuid, conn)
14631465
tm.assert_frame_equal(result, expected)
14641466
assert inserted == 2
14651467

14661468
# Cleanup
14671469
with sql.SQLDatabase(conn, need_transaction=True) as pandasSQL:
1468-
pandasSQL.drop_table("test_insert_conflict")
1470+
pandasSQL.drop_table(table_uuid)
14691471

14701472

14711473
@pytest.mark.parametrize("conn", postgresql_connectable)

0 commit comments

Comments
 (0)