Add unit test for dtype backend use for test_s3_read_parquet_table (#2307)

Author: LeonLuttenberger

awswrangler/s3/_read_parquet.py

@@ -780,7 +780,9 @@ def read_parquet_table(
     )
 
     partial_cast_function = functools.partial(
-        _data_types.cast_pandas_with_athena_types, dtype=_extract_partitions_dtypes_from_table_details(response=res)
+        _data_types.cast_pandas_with_athena_types,
+        dtype=_extract_partitions_dtypes_from_table_details(response=res),
+        dtype_backend=dtype_backend,
     )
     if _utils.is_pandas_frame(df):
         return partial_cast_function(df)

tests/unit/test_pandas_pyarrow_dtype_backend.py

@@ -26,6 +26,15 @@ def test_s3_read_parquet(path: str) -> None:
     assert_pandas_equals(df, df2)
 
 
+def test_s3_read_parquet_table(path: str, glue_database: str, glue_table: str) -> None:
+    df = get_df_dtype_backend(dtype_backend="pyarrow")
+    wr.s3.to_parquet(df=df, path=path, dataset=True, database=glue_database, table=glue_table)
+
+    df2 = wr.s3.read_parquet_table(database=glue_database, table=glue_table, dtype_backend="pyarrow")
+
+    assert_pandas_equals(df, df2)
+
+
 def test_s3_read_csv(path: str) -> None:
     df = pd.DataFrame({"id": [1, 2, 3], "val": ["foo", "boo", "bar"]})
     wr.s3.to_csv(df=df, path=f"{path}.csv", index=False)