update asvs, mypy ignores

Author: jbrockmendel

asv_bench/benchmarks/algorithms.py

@@ -199,8 +199,8 @@ class SortIntegerArray:
     params = [10**3, 10**5]
 
     def setup(self, N):
-        data = np.arange(N, dtype=float)
-        data[40] = np.nan
+        data = np.arange(N, dtype=float).astype(object)
+        data[40] = pd.NA
         self.array = pd.array(data, dtype="Int64")
 
     def time_argsort(self, N):

asv_bench/benchmarks/frame_methods.py

@@ -4,6 +4,7 @@
 import numpy as np
 
 from pandas import (
+    NA,
     DataFrame,
     Index,
     MultiIndex,
@@ -445,6 +446,8 @@ def setup(self, inplace, dtype):
             values[::2] = np.nan
             if dtype == "Int64":
                 values = values.round()
+                values = values.astype(object)
+                values[::2] = NA
             self.df = DataFrame(values, dtype=dtype)
         self.fill_values = self.df.iloc[self.df.first_valid_index()].to_dict()
 

asv_bench/benchmarks/groupby.py

@@ -689,6 +689,10 @@ def setup(self, dtype, method, with_nans):
             null_vals = vals.astype(float, copy=True)
             null_vals[::2, :] = np.nan
             null_vals[::3, :] = np.nan
+            if dtype in ["Int64", "Float64"]:
+                null_vals = null_vals.astype(object)
+                null_vals[::2, :] = NA
+                null_vals[::3, :] = NA
             df = DataFrame(null_vals, columns=list("abcde"), dtype=dtype)
             df["key"] = keys
             self.df = df

pandas/core/algorithms.py

@@ -1067,12 +1067,13 @@ def rank(
     is_datetimelike = needs_i8_conversion(values.dtype)
     if (
         isinstance(values.dtype, BaseMaskedDtype)
-        and values._hasna
+        and values._hasna  # type: ignore[union-attr]
         and values.dtype.kind in "iuf"
     ):
         # e.g. test_rank_ea_small_values
-        # TODO: bug in the object-dtype path that we would get without this special casting.
-        values = values.to_numpy(dtype=np.float64, na_value=np.nan)
+        # TODO: bug in the object-dtype path that we would get without
+        #  this special casting.
+        values = values.to_numpy(dtype=np.float64, na_value=np.nan)  # type: ignore[union-attr]
     else:
         values = _ensure_data(values)