feat(clickhouse): implement quantile/multiquantile

Author: cpcloud

ibis/backends/clickhouse/compiler/values.py

@@ -4,7 +4,7 @@
 import functools
 from functools import partial
 from operator import add, mul, sub
-from typing import Any, Literal, Mapping
+from typing import Any, Callable, Literal, Mapping
 
 import sqlglot as sg
 
@@ -180,6 +180,39 @@ def _not_all(op, **kw):
     return translate_val(ops.Not(ops.All(op.arg)), **kw)
 
 
+def _quantiles(quantiles_translator_func: Callable, func_name: str):
+    def translate(op, **kw):
+        quantile = quantiles_translator_func(op.quantile)
+        args = [_sql(translate_val(op.arg, **kw))]
+        func = func_name
+
+        if (where := op.where) is not None:
+            args.append(_sql(translate_val(where, **kw)))
+            func += "If"
+
+        return f"{func}({quantile})({', '.join(args)})"
+
+    return translate
+
+
+@translate_val.register(ops.Quantile)
+def _quantile(op, **kw):
+    def quantiles_translator_func(quantiles):
+        return _sql(translate_val(quantiles, **kw))
+
+    return _quantiles(quantiles_translator_func, func_name="quantile")(op, **kw)
+
+
+@translate_val.register(ops.MultiQuantile)
+def _multi_quantile(op, **kw):
+    def quantiles_translator_func(quantiles):
+        if not isinstance(quantiles, ops.Literal):
+            raise TypeError("ClickHouse quantile only accepts a list of Python floats")
+        return ", ".join(map(str, quantiles.value))
+
+    return _quantiles(quantiles_translator_func, func_name="quantiles")(op, **kw)
+
+
 def _agg_variance_like(func):
     variants = {"sample": f"{func}Samp", "pop": f"{func}Pop"}
 

ibis/backends/tests/test_aggregation.py

@@ -568,7 +568,6 @@ def test_reduction_ops(
                 mark.notimpl(
                     [
                         "bigquery",
-                        "clickhouse",
                         "dask",
                         "datafusion",
                         "impala",