Merge pull request #274 from JDASoftwareGroup/fix/test_arrow_compat

fix test_arrow_compat test
JDASoftwareGroup · Apr 8, 2020 · 18f8303 · 18f8303
2 parents 224ea89 + a1e8ef7
commit 18f8303
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/reference-data/arrow-compat/0.12.1.parquet b/reference-data/arrow-compat/0.12.1.parquet
diff --git a/reference-data/arrow-compat/0.13.0.parquet b/reference-data/arrow-compat/0.13.0.parquet
diff --git a/reference-data/arrow-compat/0.16.0.parquet b/reference-data/arrow-compat/0.16.0.parquet
diff --git a/tests/serialization/test_arrow_compat.py b/tests/serialization/test_arrow_compat.py
@@ -13,7 +13,7 @@
 from kartothek.serialization import ParquetSerializer
 
 
-@pytest.fixture(params=["0.12.1", "0.13.0", "0.14.1", "0.15.0"])
+@pytest.fixture(params=["0.12.1", "0.13.0", "0.14.1", "0.15.0", "0.16.0"])
 def arrow_version(request):
     yield request.param
 
@@ -49,6 +49,8 @@ def test_arrow_compat(arrow_version, reference_store, mocker):
     restored = ParquetSerializer().restore_dataframe(
         store=reference_store, key=arrow_version + ".parquet", date_as_object=True
     )
-    if arrow_version == "0.14.1" and not ARROW_LARGER_EQ_0141:
+
+    if arrow_version in ("0.14.1", "0.15.0", "0.16.0") and not ARROW_LARGER_EQ_0141:
         orig = orig.astype({"null": float})
-        pdt.assert_frame_equal(orig, restored)
+
+    pdt.assert_frame_equal(orig, restored)