Skip to content

Commit 35dcade

Browse files
authored
Add seperate code block for regular query test
1 parent 435eeeb commit 35dcade

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

tests/test_athena.py

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1099,7 +1099,7 @@ def test_start_query_execution_wait(path, glue_database, glue_table):
10991099
assert query_execution_result["QueryExecutionContext"]["Database"] == glue_database
11001100

11011101

1102-
def test_get_query_results(path, glue_database):
1102+
def test_get_query_results(path, glue_database, glue_database):
11031103

11041104
sql = (
11051105
"SELECT CAST("
@@ -1122,8 +1122,23 @@ def test_get_query_results(path, glue_database):
11221122
df_get_query_results_df_unload = wr.athena.get_query_results(query_execution_id=query_id_unload)
11231123
pd.testing.assert_frame_equal(df_get_query_results_df_unload, df_unload)
11241124

1125+
wr.catalog.delete_table_if_exists(database=glue_database, table=glue_table)
1126+
wr.s3.to_parquet(
1127+
df=get_df(),
1128+
path=path,
1129+
index=True,
1130+
use_threads=True,
1131+
dataset=True,
1132+
mode="overwrite",
1133+
database=glue_database,
1134+
table=glue_table,
1135+
partition_cols=["par0", "par1"],
1136+
)
1137+
1138+
reg_sql = f"SELECT * FROM {glue_table}"
1139+
11251140
df_regular: pd.DataFrame = wr.athena.read_sql_query(
1126-
sql=sql, database=glue_database, ctas_approach=False, unload_approach=False
1141+
sql=reg_sql, database=glue_database, ctas_approach=False, unload_approach=False
11271142
)
11281143
query_id_regular = df_regular.query_metadata["QueryExecutionId"]
11291144
df_get_query_results_df_regular = wr.athena.get_query_results(query_execution_id=query_id_regular)

0 commit comments

Comments
 (0)