feat(datafusion): implement ops.Translate

ibis-project · Feb 16, 2023 · 2fe3fc4 · 2fe3fc4
1 parent 7e61391
commit 2fe3fc4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/ibis/backends/datafusion/compiler.py b/ibis/backends/datafusion/compiler.py
@@ -471,6 +471,11 @@ def string_concat(op):
     return df.functions.concat(*map(translate, op.arg))
 
 
+@translate.register(ops.Translate)
+def string_translate(op):
+    return df.functions.translate(*map(translate, op.args))
+
+
 @translate.register(ops.RegexExtract)
 def regex_extract(op):
     arg = translate(op.arg)

diff --git a/ibis/backends/tests/test_string.py b/ibis/backends/tests/test_string.py
@@ -278,7 +278,6 @@ def test_string_col_is_unicode(alltypes, df):
             marks=pytest.mark.notimpl(
                 [
                     "clickhouse",
-                    "datafusion",
                     "duckdb",
                     "mssql",
                     "mysql",