Fixes flake in test_hypothesis_mean_agg #496

man-group · Dec 5, 2023 · 861fce4 · 861fce4
1 parent 37c5e49
commit 861fce4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/tests/unit/arcticdb/version_store/test_aggregation.py b/python/tests/unit/arcticdb/version_store/test_aggregation.py
@@ -71,6 +71,9 @@ def test_hypothesis_mean_agg(lmdb_version_store, df):
  received_df.replace(-np.inf, np.nan, inplace=True)
  received_df.replace(np.inf, np.nan, inplace=True)
 
+ expected_df.replace(-np.inf, np.nan, inplace=True)
+ expected_df.replace(np.inf, np.nan, inplace=True)
+
  assert_frame_equal(expected_df, received_df)