@@ -1709,6 +1709,7 @@ def test_athena_date_recovery(path, glue_database, glue_table):
17091709 )
17101710 assert pandas_equals (df , df2 )
17111711
1712+
17121713def 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