1 files changed, 13 insertions, 0 deletions
diff --git a/tests/dialects/test_drill.py b/tests/dialects/test_drill.py
index a196013..e41bd34 100644
--- a/tests/dialects/test_drill.py
+++ b/tests/dialects/test_drill.py
@@ -51,3 +51,16 @@ class TestDrill(Validator):
                 "drill": "SELECT * FROM table(dfs.`test_data.xlsx`(type => 'excel', sheetName => 'secondSheet'))",
             },
         )
+
+    def test_validate_pivot(self):
+        self.validate_all(
+            "SELECT * FROM (SELECT education_level, salary, marital_status, "
+            "EXTRACT(year FROM age(birth_date)) age FROM cp.`employee.json`) PIVOT (avg(salary) AS "
+            "avg_salary, avg(age) AS avg_age FOR marital_status IN ('M' married, 'S' single))",
+            write={
+                "drill": "SELECT * FROM (SELECT education_level, salary, marital_status, "
+                "EXTRACT(year FROM age(birth_date)) AS age FROM cp.`employee.json`) "
+                "PIVOT(AVG(salary) AS avg_salary, AVG(age) AS avg_age FOR marital_status "
+                "IN ('M' AS married, 'S' AS single))"
+            },
+        )