TST GH#40498 Clean up fillna benchmark tests.

Author: Mike Phung

asv_bench/benchmarks/frame_methods.py

@@ -376,11 +376,11 @@ class Fillna:
         [True, False],
         [None, "pad", "bfill"],
         [
-            "float64",
-            "float32",
             "object",
-            "Int64",
-            "Float64",
+            "int32",
+            "int64",
+            "float32",
+            "float64",
             "datetime64[ns]",
             "datetime64[ns, tz]",
             "timedelta64[ns]",
@@ -405,8 +405,7 @@ def setup(self, inplace, method, dtype):
             )
         else:
             data = np.random.randn(N, M)
-            if dtype == "Int64":
-                data = data.round()
+            data = data.astype(dtype)
             self.df = DataFrame({f"col_{i}": data[i] for i in range(M)})
             self.df[::2] = None
             self.value = dict(zip(self.df.columns, data[M])) if not method else None

asv_bench/benchmarks/series_methods.py

@@ -49,11 +49,11 @@ class Fillna:
         [True, False],
         [None, "pad", "bfill"],
         [
-            "float64",
-            "float32",
             "object",
-            "Int64",
-            "Float64",
+            "int32",
+            "int64",
+            "float32",
+            "float64",
             "datetime64[ns]",
             "datetime64[ns, tz]",
             "timedelta64[ns]",
@@ -76,8 +76,7 @@ def setup(self, inplace, method, dtype):
             self.value = dict(zip(self.ser.index, data[dtype])) if not method else None
         else:
             data = np.random.randn(N)
-            if dtype == "Int64":
-                data = data.round()
+            data = data.astype(dtype)
             self.ser = Series(data, dtype=dtype)
             self.ser[::2] = None
             self.value = dict(zip(self.ser.index, data)) if not method else None