SNOW-1355199: Fix incorrect regex used in DataFrame/Series.replace

src/snowflake/snowpark/modin/plugin/PANDAS_CHANGELOG.md

@@ -4,6 +4,7 @@
 - Fixed overriding of subclasses' property docstrings for modin issue https://github.com/modin-project/modin/issues/7113.
 - Fixed `@udf` decorator when `packages=None` are specified preventing error in `Series.apply`.
 - Fixed incorrect regex used in `Series.str.contains`.
+- Fixed incorrect regex used in `DataFrame/Series.replace`.
 
 ## 1.14.0a2 (2024-04-18)
 

src/snowflake/snowpark/modin/plugin/compiler/snowflake_query_compiler.py

@@ -10395,7 +10395,7 @@ def replace(
                 if native_pd.isna(k):
                     cond = column.is_null()
                 elif regex is True:
-                    cond = column.regexp(pandas_lit(k))
+                    cond = column.regexp(pandas_lit(f".*{k}.*"))
                     v = regexp_replace(subject=column, pattern=k, replacement=v)
                 else:
                     cond = column == k

tests/integ/modin/frame/test_replace.py

@@ -23,6 +23,9 @@ def snow_df():
     "to_replace, value",
     [
         ("one", "ONE"),  # scalar -> scalar
+        ("on", "ON"),  # scalar prefix-> scalar
+        ("ne", "NE"),  # scalar suffix-> scalar
+        ("n", "N"),  # scalar infix-> scalar
         ("one", pd.NA),  # scalar -> NULL
         ("one", None),  # scalar -> None
         (pd.NA, "ONE"),  # NULL -> scalar

tests/integ/modin/series/test_replace.py

@@ -21,6 +21,9 @@ def snow_series():
     "to_replace, value",
     [
         ("one", "ONE"),  # scalar -> scalar
+        ("on", "ON"),  # scalar prefix-> scalar
+        ("ne", "NE"),  # scalar suffix-> scalar
+        ("n", "N"),  # scalar infix-> scalar
         ("one", pd.NA),  # scalar -> NULL
         ("one", None),  # scalar -> None
         (pd.NA, "ONE"),  # NULL -> scalar