slice edge case

rapidsai · Jun 25, 2024 · 78fe267 · 78fe267
1 parent 9d70bb2
commit 78fe267
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/cudf/cudf/pylibcudf_tests/test_string_slice.py b/python/cudf/cudf/pylibcudf_tests/test_string_slice.py
@@ -76,6 +76,8 @@ def test_slice_column(
     pa_col, plc_col, pa_starts_col, plc_starts_col, pa_stops_col, plc_stops_col
 ):
     def slice_string(st, start, stop):
+        if stop < 0:
+            stop = len(st)
         return st[start:stop] if st is not None else None
 
     expected = pa.array(