From 129931bfc792fe3eb7f93935a56622e478beccd2 Mon Sep 17 00:00:00 2001 From: Jo <46752250+GeorgeSittas@users.noreply.github.com> Date: Mon, 22 May 2023 18:35:09 +0300 Subject: [PATCH] Fix(snowflake): allow 2nd argument in initcap (#1670) --- sqlglot/expressions.py | 2 +- tests/dialects/test_snowflake.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/sqlglot/expressions.py b/sqlglot/expressions.py index f4475e068c..77b8b5e0ab 100644 --- 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 index ed03615076..58b2a2dafa 100644 --- 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'")