Add test arguments for test_mask_na

Co-authored by: rhshadrach <[email protected]>

Author: sanggon6107

pandas/tests/series/indexing/test_mask.py

@@ -2,7 +2,6 @@
 import pytest
 
 from pandas import (
-    Int64Dtype,
     Series,
 )
 import pandas._testing as tm
@@ -72,10 +71,11 @@ def test_mask_inplace():
     tm.assert_series_equal(rs, s.mask(cond, -s))
 
 
-def test_mask_na():
+@pytest.mark.parametrize("dtype", ["Int64", "int64[pyarrow]"])
+def test_mask_na(dtype):
     # We should not be filling pd.NA. See GH#60729
-    series = Series([None, 1, 2, None, 3, 4, None], dtype=Int64Dtype())
+    series = Series([None, 1, 2, None, 3, 4, None], dtype=dtype)
     result = series.mask(series <= 2, -99)
-    expected = Series([None, -99, -99, None, 3, 4, None], dtype=Int64Dtype())
+    expected = Series([None, -99, -99, None, 3, 4, None], dtype=dtype)
 
     tm.assert_series_equal(result, expected)