Skip to content

Commit 016a082

Browse files
krzysztof-kwittcpcloud
authored andcommitted
feat(datafusion): implement ops.StrRight
1 parent 8099014 commit 016a082

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

ibis/backends/datafusion/compiler.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -486,6 +486,11 @@ def string_starts_with(op):
486486
return df.functions.starts_with(translate(op.arg), translate(op.start))
487487

488488

489+
@translate.register(ops.StrRight)
490+
def string_right(op):
491+
return df.functions.right(translate(op.arg), translate(op.nchars))
492+
493+
489494
@translate.register(ops.RegexExtract)
490495
def regex_extract(op):
491496
arg = translate(op.arg)

ibis/backends/tests/test_string.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -512,9 +512,6 @@ def test_string_col_is_unicode(alltypes, df):
512512
lambda t: t.date_string_col.right(2),
513513
lambda t: t.date_string_col.str[-2:],
514514
id="right",
515-
marks=pytest.mark.notimpl(
516-
["datafusion"], raises=com.OperationNotDefinedError
517-
),
518515
),
519516
param(
520517
lambda t: t.date_string_col[1:3],

0 commit comments

Comments
 (0)