feat(polars): add StringFind operation

Author: IndexSeek

ibis/backends/polars/compiler.py

@@ -1555,3 +1555,12 @@ def visit_ArrayIntersect(op, **kw):
     left = translate(op.left, **kw)
     right = translate(op.right, **kw)
     return left.list.set_intersection(right)
+
+
+@translate.register(ops.StringFind)
+def visit_StringFind(op, **kw):
+    arg = translate(op.arg, **kw)
+    start = translate(op.start, **kw) if op.start is not None else 0
+    end = translate(op.end, **kw) if op.end is not None else None
+    expr = arg.str.slice(start, end).str.find(_literal_value(op.substr), literal=True)
+    return pl.when(expr.is_null()).then(-1).otherwise(expr + start)

ibis/backends/tests/test_string.py

@@ -406,15 +406,11 @@ def uses_java_re(t):
             lambda t: t.string_col.find("a"),
             lambda t: t.string_col.str.find("a"),
             id="find",
-            marks=pytest.mark.notimpl(["polars"], raises=com.OperationNotDefinedError),
         ),
         param(
             lambda t: t.date_string_col.find("13", 3),
             lambda t: t.date_string_col.str.find("13", 3),
             id="find_start",
-            marks=[
-                pytest.mark.notimpl(["polars"], raises=com.OperationNotDefinedError),
-            ],
         ),
         param(
             lambda t: t.string_col.lpad(10, "a"),
@@ -605,6 +601,8 @@ def test_string(backend, alltypes, df, result_func, expected_func):
     result = expr.execute()
 
     expected = backend.default_series_rename(expected_func(df))
+    result.to_csv("result.csv", index=False)
+    expected.to_csv("expected.csv", index=False)
     backend.assert_series_equal(result, expected)
 
 
@@ -1084,7 +1082,6 @@ def string_temp_table(backend, con):
             lambda t: t.string_col.find("123"),
             lambda t: t.str.find("123"),
             id="find",
-            marks=pytest.mark.notimpl("polars", raises=com.OperationNotDefinedError),
         ),
         param(
             lambda t: t.string_col.rpad(4, "-"),