feat(clickhouse): implement ops.Capitalize

ibis-project · Feb 5, 2023 · 914c64c · 914c64c
1 parent 104aeed
commit 914c64c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/ibis/backends/clickhouse/compiler/values.py b/ibis/backends/clickhouse/compiler/values.py
@@ -653,6 +653,12 @@ def _string_ilike(op, **kw):
     return f"lower({arg}) LIKE lower({pattern})"
 
 
+@translate_val.register(ops.Capitalize)
+def _string_capitalize(op, **kw):
+    arg = translate_val(op.arg, **kw)
+    return f"CONCAT(UPPER(SUBSTR({arg}, 1, 1)), SUBSTR({arg}, 2))"
+
+
 @translate_val.register(ops.GroupConcat)
 def _group_concat(op, **kw):
     arg = translate_val(op.arg, **kw)

diff --git a/ibis/backends/tests/test_string.py b/ibis/backends/tests/test_string.py
@@ -268,7 +268,7 @@ def test_string_col_is_unicode(alltypes, df):
             lambda t: t.string_col.capitalize(),
             lambda t: t.string_col.str.capitalize(),
             id='capitalize',
-            marks=pytest.mark.notimpl(["clickhouse", "mssql"]),
+            marks=pytest.mark.notimpl(["mssql"]),
         ),
         param(
             lambda t: t.date_string_col.substr(2, 3),