feat(clickhouse): implement ArrayColumn

Author: cpcloud

ibis/backends/clickhouse/registry.py

@@ -609,6 +609,11 @@ def compile(translator, expr):
     return compile
 
 
+def _array_column(t, expr):
+    args = ", ".join(map(t.translate, expr.op().cols))
+    return f"[{args}]"
+
+
 # TODO: clickhouse uses different string functions
 #       for ascii and utf-8 encodings,
 
@@ -762,6 +767,7 @@ def compile(translator, expr):
     ops.Degrees: _unary("degrees"),
     ops.Radians: _unary("radians"),
     ops.Strftime: _fixed_arity("formatDateTime", 2),
+    ops.ArrayColumn: _array_column,
 }
 
 

ibis/backends/tests/test_array.py

@@ -10,16 +10,14 @@
 pytestmark = pytest.mark.never(["sqlite", "mysql"], reason="No array support")
 
 
-@pytest.mark.notimpl(["impala", "clickhouse", "datafusion"])
+@pytest.mark.notimpl(["impala", "datafusion"])
 def test_array_column(backend, alltypes, df):
     expr = ibis.array([alltypes['double_col'], alltypes['double_col']])
     assert isinstance(expr, ir.ArrayColumn)
 
     result = expr.execute()
     expected = df.apply(
-        lambda row: np.array(
-            [row['double_col'], row['double_col']], dtype=object
-        ),
+        lambda row: [row['double_col'], row['double_col']],
         axis=1,
     )
     backend.assert_series_equal(result, expected, check_names=False)