Improve tests for replace

Author: sfc-gh-jkew

tests/integ/modin/frame/test_replace.py

@@ -22,28 +22,45 @@ def snow_df():
         {"col1": ["one", "two", "three", pd.NA], "col2": ["abc", "pqr", "xyz", None]}
     )
 
+
 @sql_count_checker(query_count=4)
-def test_replace_mauricio():
+def test_replace_mauricio_dataframe():
     test_sdf = pd.DataFrame(
-        {"col1": ["one", "two", "two", "three", "two", "four"], "col2": ["abc", "pqr", "xyz", None, "pqr", "xyz",]}
+        {
+            "col1": ["one", "two", "two", "three", "two", "four"],
+            "col2": [
+                "abc",
+                "pqr",
+                "xyz",
+                None,
+                "pqr",
+                "xyz",
+            ],
+        }
     )
-    test_ndf = test_sdf.to_pandas()  
-    eval_snowpark_pandas_result(
-        test_sdf,
-        test_ndf,
-        lambda x : x
-    )  
-    test_sdf["col1"] = test_sdf["col1"].replace(["two"], ["a"])
-    print(test_sdf)
-    
-    # Fails in native pandas HERE
-    test_ndf["col1"] = test_ndf["col1"].replace(["two"], ["a"])
-    print(test_ndf)
+    test_ndf = test_sdf.to_pandas()
+    test_mdf = test_sdf.move_to("Pandas")
+    assert test_mdf.get_backend() == "Pandas"
+
     eval_snowpark_pandas_result(
-        test_sdf,
-        test_ndf,
-        lambda x : x
+        test_sdf, test_ndf, lambda df: df["col1"].replace(["two"], ["a"])
     )
+    snow_result = test_sdf["col1"].replace(["two"], ["a"])
+    modin_result = test_mdf["col1"].replace(["two"], ["a"])
+    assert modin_result.to_pandas().equals(snow_result.to_pandas())
+
+
+@sql_count_checker(query_count=4, join_count=2)
+def test_replace_mauricio_series():
+
+    input_dict = pd.Series({"a": "apple", "socks": "socks are darned", "b": "bee"})
+    input_dict_native = input_dict.to_pandas()
+    input_dict_pandas_modin = input_dict.move_to("Pandas")
+    assert input_dict_pandas_modin.get_backend() == "Pandas"
+    expected = input_dict_native["socks"].replace("darned", "clean")
+    assert input_dict["socks"].replace("darned", "clean") == expected
+    assert input_dict_pandas_modin["socks"].replace("darned", "clean") == expected
+
 
 @pytest.mark.parametrize(
     "to_replace, value",