TST GH#40498 Return Int64 and Float64 to asv benchmark and update setups.

Author: Mike Phung

asv_bench/benchmarks/frame_methods.py

@@ -379,8 +379,10 @@ class Fillna:
             "object",
             "int32",
             "int64",
+            "Int64",
             "float32",
             "float64",
+            "Float64",
             "datetime64[ns]",
             "datetime64[ns, tz]",
             "timedelta64[ns]",
@@ -399,16 +401,23 @@ def setup(self, inplace, method, dtype):
                 "timedelta64[ns]": timedelta_range(start="1 day", periods=N, freq="1D"),
             }
             self.df = DataFrame({f"col_{i}": data[dtype] for i in range(M)})
-            self.df[::2] = None
             self.value = (
-                dict(zip(self.df.columns, data[dtype][:M])) if not method else None
+                dict(zip(self.df.columns, self.df.sample().iloc[0]))
+                if not method
+                else None
             )
+            self.df[::2] = None
         else:
             data = np.random.randn(N, M)
-            data = data.astype(dtype)
-            self.df = DataFrame({f"col_{i}": data[i] for i in range(M)})
+            if dtype in ["int32", "int64", "Int64"]:
+                data = data.round()
+            self.df = DataFrame({f"col_{i}": data[i] for i in range(M)}, dtype=dtype)
+            self.value = (
+                dict(zip(self.df.columns, self.df.sample().iloc[0]))
+                if not method
+                else None
+            )
             self.df[::2] = None
-            self.value = dict(zip(self.df.columns, data[M])) if not method else None
 
     def time_frame_fillna(self, inplace, method, dtype):
         self.df.fillna(value=self.value, inplace=inplace, method=method)

asv_bench/benchmarks/series_methods.py

@@ -65,8 +65,10 @@ class Fillna:
             "object",
             "int32",
             "int64",
+            "Int64",
             "float32",
             "float64",
+            "Float64",
             "datetime64[ns]",
             "datetime64[ns, tz]",
             "timedelta64[ns]",
@@ -85,14 +87,19 @@ def setup(self, inplace, method, dtype):
                 "timedelta64[ns]": timedelta_range(start="1 day", periods=N, freq="1D"),
             }
             self.ser = Series(data[dtype])
+            self.value = (
+                dict(zip(self.ser.index, self.ser.values)) if not method else None
+            )
             self.ser[::2] = None
-            self.value = dict(zip(self.ser.index, data[dtype])) if not method else None
         else:
             data = np.random.randn(N)
-            data = data.astype(dtype)
+            if dtype in ["int32", "int64", "Int64"]:
+                data = data.round()
             self.ser = Series(data, dtype=dtype)
+            self.value = (
+                dict(zip(self.ser.index, self.ser.values)) if not method else None
+            )
             self.ser[::2] = None
-            self.value = dict(zip(self.ser.index, data)) if not method else None
 
     def time_series_fillna(self, inplace, method, dtype):
         self.ser.fillna(value=self.value, inplace=inplace, method=method)