feat(snowflake)!: annotate types for BITNOT (#6234)

Author: fivetran-felixhuang

sqlglot/dialects/snowflake.py

@@ -635,6 +635,7 @@ class Parser(parser.Parser):
                 step=seq_get(args, 2),
             ),
             "ARRAY_SORT": exp.SortArray.from_arg_list,
+            "BITNOT": lambda args: exp.BitwiseNot(this=seq_get(args, 0)),
             "BITXOR": _build_bitwise(exp.BitwiseXor, "BITXOR"),
             "BIT_XOR": _build_bitwise(exp.BitwiseXor, "BITXOR"),
             "BITOR": _build_bitwise(exp.BitwiseOr, "BITOR"),

tests/dialects/test_snowflake.py

@@ -1345,6 +1345,7 @@ def test_snowflake(self):
                 "snowflake": "EDITDISTANCE(col1, col2, 3)",
             },
         )
+        self.validate_identity("SELECT BITNOT(a)")
         self.validate_identity("SELECT BITOR(a, b)")
         self.validate_identity("SELECT BIT_OR(a, b)", "SELECT BITOR(a, b)")
         self.validate_identity("SELECT BITOR(a, b, 'LEFT')")

tests/fixtures/optimizer/annotate_functions.sql

@@ -1655,6 +1655,14 @@ INT;
 BIT_LENGTH(tbl.bin_col);
 INT;
 
+# dialect: snowflake
+BITNOT(5);
+INT;
+
+# dialect: snowflake
+BITNOT(tbl.bin_col);
+BINARY;
+
 # dialect: snowflake
 BITOR(2, 4);
 INT;