Fix(snowflake): allow 2nd argument in initcap (#1670)

tobymao · May 22, 2023 · 129931b · 129931b
1 parent 64fdca4
commit 129931b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/sqlglot/expressions.py b/sqlglot/expressions.py
@@ -3931,7 +3931,7 @@ class IfNull(Func):
 
 
 class Initcap(Func):
-    pass
+    arg_types = {"this": True, "expression": False}
 
 
 class JSONKeyValue(Expression):

diff --git a/tests/dialects/test_snowflake.py b/tests/dialects/test_snowflake.py
@@ -6,6 +6,7 @@ class TestSnowflake(Validator):
     dialect = "snowflake"
 
     def test_snowflake(self):
+        self.validate_identity("INITCAP('iqamqinterestedqinqthisqtopic', 'q')")
         self.validate_identity("CAST(x AS GEOMETRY)")
         self.validate_identity("OBJECT_CONSTRUCT(*)")
         self.validate_identity("SELECT TO_DATE('2019-02-28') + INTERVAL '1 day, 1 year'")
-Original file line number
+Diff line change
@@ Expand Up / @@ -3931,7 +3931,7 @@ class IfNull(Func): @@
     class Initcap(Func):
-        pass
+        arg_types = {"this": True, "expression": False}
     class JSONKeyValue(Expression):
@@ Expand Down @@