feat(datafusion): implement ops.StartsWith

ibis-project · Feb 16, 2023 · 8099014 · 8099014
1 parent b1d7672
commit 8099014
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ibis/backends/datafusion/compiler.py b/ibis/backends/datafusion/compiler.py
@@ -481,6 +481,11 @@ def string_ascii(op):
     return df.functions.ascii(translate(op.arg))
 
 
+@translate.register(ops.StartsWith)
+def string_starts_with(op):
+    return df.functions.starts_with(translate(op.arg), translate(op.start))
+
+
 @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
@@ -423,7 +423,7 @@ def test_string_col_is_unicode(alltypes, df):
             id='startswith-simple',
             marks=[
                 pytest.mark.notimpl(
-                    ["dask", "datafusion"],
+                    ["dask"],
                     raises=com.OperationNotDefinedError,
                 ),
                 pytest.mark.broken(