moved new tests to strings/test_find_replace.py

Author: Khemkaran

pandas/core/strings/object_array.py

@@ -248,14 +248,15 @@ def rep(x, r):
 
     def _str_match(
         self,
-        pat: str,
+        pat,
         case: bool = True,
         flags: int = 0,
         na: Scalar | lib.NoDefault = lib.no_default,
     ):
         if not case:
             flags |= re.IGNORECASE
-
+        if isinstance(pat, re.Pattern):
+            pat = pat.pattern
         regex = re.compile(pat, flags=flags)
 
         f = lambda x: regex.match(x) is not None
@@ -270,7 +271,8 @@ def _str_fullmatch(
     ):
         if not case:
             flags |= re.IGNORECASE
-
+        if isinstance(pat, re.Pattern):
+            pat = pat.pattern
         regex = re.compile(pat, flags=flags)
 
         f = lambda x: regex.fullmatch(x) is not None

pandas/tests/extension/test_arrow.py

@@ -1857,10 +1857,6 @@ def test_str_repeat():
         ["ab", False, True, [True, True]],
         ["a[a-z]{1}", False, None, [True, None]],
         ["A[a-z]{1}", True, None, [False, None]],
-        # GH#61952
-        [re.compile(r"ab"), False, None, [True, None]],
-        [re.compile(r"Abc"), True, None, [False, None]],
-        [re.compile(r"a[a-z]{1}"), False, None, [True, None]],
     ],
 )
 def test_str_match(pat, case, na, exp):
@@ -1884,10 +1880,6 @@ def test_str_match(pat, case, na, exp):
         ["abc\\$", False, None, [False, True, False, None]],
         ["Abc$", True, None, [False, False, False, None]],
         ["Abc\\$", True, None, [False, False, False, None]],
-        # GH#61952
-        [re.compile(r"abc"), False, None, [True, True, False, None]],
-        [re.compile(r"abc$"), False, None, [True, False, False, None]],
-        [re.compile(r"a[a-z]{2}"), False, None, [True, True, False, None]],
     ],
 )
 def test_str_fullmatch(pat, case, na, exp):

pandas/tests/strings/test_find_replace.py

@@ -818,6 +818,16 @@ def test_match_case_kwarg(any_string_dtype):
     tm.assert_series_equal(result, expected)
 
 
+def test_match_compiled_regex(any_string_dtype):
+    values = Series(["ab", "AB", "abc", "ABC"], dtype=any_string_dtype)
+    result = values.str.match(re.compile(r"ab"), case=False)
+    expected_dtype = (
+        np.bool_ if is_object_or_nan_string_dtype(any_string_dtype) else "boolean"
+    )
+    expected = Series([True, True, True, True], dtype=expected_dtype)
+    tm.assert_series_equal(result, expected)
+
+
 # --------------------------------------------------------------------------------------
 # str.fullmatch
 # --------------------------------------------------------------------------------------
@@ -887,6 +897,16 @@ def test_fullmatch_case_kwarg(any_string_dtype):
     tm.assert_series_equal(result, expected)
 
 
+def test_fullmatch_compiled_regex(any_string_dtype):
+    values = Series(["ab", "AB", "abc", "ABC"], dtype=any_string_dtype)
+    result = values.str.fullmatch(re.compile(r"ab"), case=False)
+    expected_dtype = (
+        np.bool_ if is_object_or_nan_string_dtype(any_string_dtype) else "boolean"
+    )
+    expected = Series([True, True, False, False], dtype=expected_dtype)
+    tm.assert_series_equal(result, expected)
+
+
 # --------------------------------------------------------------------------------------
 # str.findall
 # --------------------------------------------------------------------------------------