diff --git a/sqlglot/dialects/snowflake.py b/sqlglot/dialects/snowflake.py index 2e9ba304e7..f1ce7f0488 100644 --- a/sqlglot/dialects/snowflake.py +++ b/sqlglot/dialects/snowflake.py @@ -573,6 +573,7 @@ class Snowflake(Dialect): exp.Sin, exp.Tan, exp.Asin, + exp.Atan2, exp.Atanh, exp.Cbrt, }, diff --git a/tests/dialects/test_snowflake.py b/tests/dialects/test_snowflake.py index 6fed557291..1cc8c1df05 100644 --- a/tests/dialects/test_snowflake.py +++ b/tests/dialects/test_snowflake.py @@ -60,6 +60,7 @@ def test_snowflake(self): self.validate_identity("SELECT SOUNDEX_P123(column_name)") self.validate_identity("SELECT ABS(x)") self.validate_identity("SELECT ASIN(0.5)") + self.validate_identity("SELECT ATAN2(0.5, 0.3)") self.validate_identity("SELECT ATANH(0.5)") self.validate_identity("SELECT CBRT(27.0)") self.validate_identity("SELECT SIGN(x)") diff --git a/tests/fixtures/optimizer/annotate_functions.sql b/tests/fixtures/optimizer/annotate_functions.sql index 89502e039e..e88f205997 100644 --- a/tests/fixtures/optimizer/annotate_functions.sql +++ b/tests/fixtures/optimizer/annotate_functions.sql @@ -1547,6 +1547,10 @@ DOUBLE; ASIN(tbl.double_col); DOUBLE; +# dialect: snowflake +ATAN2(tbl.double_col, tbl.double_col); +DOUBLE; + # dialect: snowflake ATANH(tbl.double_col); DOUBLE;