Support for empty dataframe for Pandas.read_sql_athena(ctas_approach=True)

Author: igorborgest

awswrangler/pandas.py

@@ -582,6 +582,8 @@ def _read_sql_athena_ctas(self,
         self._session.glue.delete_table_if_exists(database=database, table=name)
         manifest_path: str = f"{s3_output}/tables/{query_id}-manifest.csv"
         paths: List[str] = self._session.athena.extract_manifest_paths(path=manifest_path)
+        if not paths:
+            return pd.DataFrame()
         logger.debug(f"paths: {paths}")
         return self.read_parquet(path=paths,
                                  procs_cpu_bound=procs_cpu_bound,

testing/test_awswrangler/test_pandas.py

@@ -2020,3 +2020,17 @@ def test_read_csv_prefix_iterator(bucket, sample, row_num):
         total_count += count
     wr.s3.delete_listed_objects(objects_paths=paths)
     assert total_count == row_num * n
+
+
+@pytest.mark.parametrize("ctas_approach", [False, True])
+def test_read_sql_athena_empty(ctas_approach):
+    sql = """
+        WITH dataset AS (
+          SELECT 0 AS id
+        )
+        SELECT id
+        FROM dataset
+        WHERE id != 0
+    """
+    df = wr.pandas.read_sql_athena(sql=sql, ctas_approach=ctas_approach)
+    print(df)