feat(optimizer): Annotate type for snowflake SIGN and ABS functions (#6025)

* feat(optimizer): Annotate type for snowflake SIGN and ABS functions

* fix: keeping ABS generic

Author: fivetran-amrutabhimsenayachit

tests/dialects/test_snowflake.py

@@ -47,6 +47,8 @@ def test_snowflake(self):
         self.validate_identity("SELECT RPAD(tbl.bin_col, 10)")
         self.validate_identity("SELECT SOUNDEX(column_name)")
         self.validate_identity("SELECT SOUNDEX_P123(column_name)")
+        self.validate_identity("SELECT ABS(x)")
+        self.validate_identity("SELECT SIGN(x)")
         self.validate_identity("SELECT JAROWINKLER_SIMILARITY('hello', 'world')")
         self.validate_identity("SELECT TRANSLATE(column_name, 'abc', '123')")
         self.validate_identity("SELECT UNICODE(column_name)")

tests/fixtures/optimizer/annotate_functions.sql

@@ -1535,6 +1535,14 @@ STRUCT<str_col STRING>;
 -- Snowflake
 --------------------------------------
 
+# dialect: snowflake
+ABS(tbl.bigint_col);
+BIGINT;
+
+# dialect: snowflake
+ABS(tbl.double_col);
+DOUBLE;
+
 # dialect: snowflake
 AI_AGG('foo', 'bar');
 VARCHAR;
@@ -2159,6 +2167,10 @@ VARCHAR;
 SIN(tbl.double_col);
 DOUBLE;
 
+# dialect: snowflake
+SIGN(tbl.double_col);
+INT;
+
 # dialect: snowflake
 SOUNDEX(tbl.str_col);
 VARCHAR;