Fix fillna error in CoW chain

coiled · Sep 21, 2023 · 77adb1b · 77adb1b
1 parent 565617b
commit 77adb1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/benchmarks/test_dataframe.py b/tests/benchmarks/test_dataframe.py
@@ -88,7 +88,7 @@ def test_dataframe_cow_chain(small_client):
     result = (
         df.rename(columns={1: 1000})
         .replace("x", "xxx")
-        .fillna(100)
+        .fillna({i: 100 for i in range(10, 70)})
         .astype({50: "float"})
         .loc[:, slice(2, 100)]
     )