Skip to content

Commit b1c204c

Browse files
committed
Parallelize test_sql.py - removed pytest.mark.single_cpu added pytest.mark.db
1 parent 19917fd commit b1c204c

File tree

1 file changed

+5
-22
lines changed

1 file changed

+5
-22
lines changed

pandas/tests/io/test_sql.py

Lines changed: 5 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@
5959
pytest.mark.filterwarnings(
6060
"ignore:Passing a BlockManager to DataFrame:DeprecationWarning"
6161
),
62-
pytest.mark.single_cpu,
62+
pytest.mark.db,
6363
]
6464

6565

@@ -585,24 +585,6 @@ def get_all_tables(conn):
585585
return inspect(conn).get_table_names()
586586

587587

588-
def filter_get_all_tables(conn, extract_this_value):
589-
tables = get_all_tables(conn)
590-
return_val = []
591-
for t in tables:
592-
if t in extract_this_value:
593-
return_val.append(t)
594-
return return_val
595-
596-
def filter_get_all_views(conn, extract_this_value):
597-
views = get_all_views(conn)
598-
return_val = []
599-
600-
for v in views:
601-
if v in extract_this_value:
602-
return_val.append(v)
603-
return return_val
604-
605-
606588
def drop_table(
607589
table_name: str,
608590
conn: sqlite3.Connection | sqlalchemy.engine.Engine | sqlalchemy.engine.Connection,
@@ -620,9 +602,10 @@ def drop_table(
620602
else:
621603
import sqlalchemy
622604
# Better matching for dialect string literal tables
623-
quoted_table_name = conn.engine.dialect.identifier_preparer.quote_identifier(
605+
quoted_table_name = (
606+
conn.engine.dialect.identifier_preparer.quote_identifier(
624607
table_name
625-
)
608+
))
626609
stmt = sqlalchemy.text(f"DROP TABLE IF EXISTS {quoted_table_name}")
627610

628611
if isinstance(conn, sqlalchemy.Engine):
@@ -4900,4 +4883,4 @@ def test_xsqlite_if_exists(connect_and_uuid):
49004883
(4, "D"),
49014884
(5, "E"),
49024885
]
4903-
drop_table(table_uuid, conn)
4886+
drop_table(table_uuid, conn)

0 commit comments

Comments
 (0)