fix(backends): fix capitalize to lowercase subsequent characters

.gitignore

@@ -39,6 +39,15 @@ dist
 .coverage
 coverage.xml
 
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
 # OS generated files
 .directory
 .gdb_history

ibis/backends/base/sql/alchemy/registry.py

@@ -553,7 +553,8 @@ class array_filter(FunctionElement):
     # string
     ops.Capitalize: unary(
         lambda arg: sa.func.concat(
-            sa.func.upper(sa.func.substr(arg, 1, 1)), sa.func.substr(arg, 2)
+            sa.func.upper(sa.func.substr(arg, 1, 1)),
+            sa.func.lower(sa.func.substr(arg, 2)),
         )
     ),
     ops.LPad: fixed_arity(sa.func.lpad, 3),

ibis/backends/bigquery/registry.py

@@ -577,7 +577,7 @@ def _array_slice(t, op):
 
 def _capitalize(t, op):
     arg = t.translate(op.arg)
-    return f"CONCAT(UPPER(SUBSTR({arg}, 1, 1)), SUBSTR({arg}, 2))"
+    return f"CONCAT(UPPER(SUBSTR({arg}, 1, 1)), LOWER(SUBSTR({arg}, 2)))"
 
 
 def _nth_value(t, op):

ibis/backends/clickhouse/compiler/values.py

@@ -675,7 +675,7 @@ def _string_ilike(op, **kw):
 @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))"
+    return f"CONCAT(UPPER(SUBSTR({arg}, 1, 1)), LOWER(SUBSTR({arg}, 2)))"
 
 
 @translate_val.register(ops.GroupConcat)

ibis/backends/mssql/registry.py

@@ -122,7 +122,7 @@ def _timestamp_truncate(t, op):
         ops.Capitalize: unary(
             lambda arg: sa.func.concat(
                 sa.func.upper(sa.func.substring(arg, 1, 1)),
-                sa.func.substring(arg, 2, sa.func.datalength(arg) - 1),
+                sa.func.lower(sa.func.substring(arg, 2, sa.func.datalength(arg) - 1)),
             )
         ),
         ops.LStrip: unary(sa.func.ltrim),

ibis/backends/tests/test_string.py

@@ -974,7 +974,7 @@ def test_parse_url(con, result_func, expected):
 
 
 def test_capitalize(con):
-    s = ibis.literal("abc")
+    s = ibis.literal("aBc")
     expected = "Abc"
     expr = s.capitalize()
     assert con.execute(expr) == expected