escape one check on Pandas 1

man-group · Nov 6, 2024 · 4b36542 · 4b36542
1 parent 26c0fa8
commit 4b36542
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/python/tests/integration/arcticdb/test_read_batch_more.py b/python/tests/integration/arcticdb/test_read_batch_more.py
@@ -304,9 +304,14 @@ def generate_mixed_dataframe(num_rows: int, seed=0):
     assert_frame_equal_row_range_fix(dfqapplied, batch[5].data)
     assert metadata3 == batch[5].metadata
     assert metadata3 == lib.read_metadata(symbol2).metadata
-    # Filter fload and string condition
-    dfqapplied = df1_all.query(qdf4)
-    assert_frame_equal_row_range_fix(dfqapplied, batch[7].data)
+
+    # Last test will fail on Pandas 1 due to fact that
+    # when empty df is returned object type columns will have type float64
+    v = pd.__version__
+    if (v[0] == 2):
+        # Filter fload and string condition
+        dfqapplied = df1_all.query(qdf4)
+        assert_frame_equal_row_range_fix(dfqapplied, batch[7].data)
 
 
 def test_read_batch_multiple_wrong_things_at_once(arctic_library):