Skip to content

Commit 3d2588c

Browse files
committed
chore(ci): run ruff on athena tests file
1 parent 32d3145 commit 3d2588c

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

tests/unit/test_athena.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1709,6 +1709,7 @@ def test_athena_date_recovery(path, glue_database, glue_table):
17091709
)
17101710
assert pandas_equals(df, df2)
17111711

1712+
17121713
def test_start_query_executions_ids_and_results(path, glue_database, glue_table):
17131714
# Prepare table
17141715
wr.s3.to_parquet(
@@ -1740,9 +1741,7 @@ def test_start_query_executions_ids_and_results(path, glue_database, glue_table)
17401741
assert all("Status" in r for r in results)
17411742

17421743
# Case 3: Parallel execution with threads
1743-
results_parallel = wr.athena.start_query_executions(
1744-
sqls=sqls, database=glue_database, wait=True, use_threads=True
1745-
)
1744+
results_parallel = wr.athena.start_query_executions(sqls=sqls, database=glue_database, wait=True, use_threads=True)
17461745
assert isinstance(results_parallel, list)
17471746
assert all(isinstance(r, dict) for r in results_parallel)
17481747

@@ -1763,9 +1762,7 @@ def test_start_query_executions_as_iterator(path, glue_database, glue_table):
17631762
sqls = [f"SELECT * FROM {glue_table} LIMIT 1"]
17641763

17651764
# Case: as_iterator=True should return a generator-like object
1766-
qids_iter = wr.athena.start_query_executions(
1767-
sqls=sqls, database=glue_database, wait=False, as_iterator=True
1768-
)
1765+
qids_iter = wr.athena.start_query_executions(sqls=sqls, database=glue_database, wait=False, as_iterator=True)
17691766
assert not isinstance(qids_iter, list)
17701767
qids = list(qids_iter)
17711768
assert len(qids) == 1

0 commit comments

Comments
 (0)