feat(bigquery): implement ops.Capitalize

ibis-project · Jan 4, 2023 · 5757bb0 · 5757bb0
1 parent 49414ef
commit 5757bb0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/ibis/backends/bigquery/registry.py b/ibis/backends/bigquery/registry.py
@@ -508,6 +508,10 @@ def _array_slice(t, op):
     )
 
 
+def _capitalize(t, op):
+    return f"CONCAT(UPPER(SUBSTR({t.translate(op.arg)}, 1, 1)), SUBSTR({t.translate(op.arg)}, 2))"
+
+
 OPERATION_REGISTRY = {
     **operation_registry,
     # Literal
@@ -567,6 +571,7 @@ def _array_slice(t, op):
     ops.StringFind: _string_find,
     ops.Substring: _string_substring,
     ops.StrRight: _string_right,
+    ops.Capitalize: _capitalize,
     ops.Repeat: fixed_arity("REPEAT", 2),
     ops.RegexSearch: _regex_search,
     ops.RegexExtract: _regex_extract,
@@ -639,7 +644,6 @@ def _array_slice(t, op):
 _invalid_operations = {
     ops.Translate,
     ops.FindInSet,
-    ops.Capitalize,
     ops.DateDiff,
     ops.TimestampDiff,
 }

diff --git a/ibis/backends/tests/test_string.py b/ibis/backends/tests/test_string.py
@@ -273,7 +273,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(["bigquery", "clickhouse", "duckdb", "mssql"]),
+            marks=pytest.mark.notimpl(["clickhouse", "duckdb", "mssql"]),
         ),
         param(
             lambda t: t.date_string_col.substr(2, 3),