feat(bigquery): implement ops.Clip

Author: krzysztof-kwitt

ibis/backends/bigquery/registry.py

@@ -512,6 +512,18 @@ def _capitalize(t, op):
     return f"CONCAT(UPPER(SUBSTR({t.translate(op.arg)}, 1, 1)), SUBSTR({t.translate(op.arg)}, 2))"
 
 
+def _clip(t, op):
+    arg = t.translate(op.arg)
+
+    if (upper := op.upper) is not None:
+        arg = f"LEAST({t.translate(upper)}, {arg})"
+
+    if (lower := op.lower) is not None:
+        arg = f"GREATEST({t.translate(lower)}, {arg})"
+
+    return arg
+
+
 OPERATION_REGISTRY = {
     **operation_registry,
     # Literal
@@ -533,6 +545,7 @@ def _capitalize(t, op):
     ops.Floor: compiles_floor,
     ops.Modulus: fixed_arity("MOD", 2),
     ops.Sign: unary("SIGN"),
+    ops.Clip: _clip,
     # Temporal functions
     ops.Date: unary("DATE"),
     ops.DateFromYMD: fixed_arity("DATE", 3),

ibis/backends/tests/test_numeric.py

@@ -583,7 +583,7 @@ def test_random(con):
         ),
     ],
 )
-@pytest.mark.notimpl(["bigquery", "datafusion", "impala"])
+@pytest.mark.notimpl(["datafusion", "impala"])
 def test_clip(alltypes, df, ibis_func, pandas_func):
     result = ibis_func(alltypes.int_col).execute()
     expected = pandas_func(df.int_col).astype(result.dtype)