Skip fastparquet tests for numpy 2 (#11103)

dask · May 6, 2024 · 7b2a47e · 7b2a47e
1 parent 95f0704
commit 7b2a47e
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/dask/bytes/tests/test_s3.py b/dask/bytes/tests/test_s3.py
@@ -461,6 +461,12 @@ def test_parquet(s3, engine, s3so, metadata_file):
     pd = pytest.importorskip("pandas")
     np = pytest.importorskip("numpy")
 
+    from dask.array.numpy_compat import NUMPY_GE_200
+
+    if NUMPY_GE_200 and engine == "fastparquet":
+        # https://github.com/dask/fastparquet/issues/923
+        pytest.skip("fastparquet doesn't work with Numpy 2")
+
     url = "s3://%s/test.parquet" % test_bucket_name
     data = pd.DataFrame(
         {
@@ -556,6 +562,11 @@ def test_parquet_append(s3, engine, s3so):
     dd = pytest.importorskip("dask.dataframe")
     pd = pytest.importorskip("pandas")
     np = pytest.importorskip("numpy")
+    from dask.array.numpy_compat import NUMPY_GE_200
+
+    if NUMPY_GE_200 and engine == "fastparquet":
+        # https://github.com/dask/fastparquet/issues/923
+        pytest.skip("fastparquet doesn't work with Numpy 2")
     if dd._dask_expr_enabled():
         pytest.skip("need convert string option")
 
@@ -612,6 +623,12 @@ def test_parquet_wstoragepars(s3, s3so, engine):
     pd = pytest.importorskip("pandas")
     np = pytest.importorskip("numpy")
 
+    from dask.array.numpy_compat import NUMPY_GE_200
+
+    if NUMPY_GE_200 and engine == "fastparquet":
+        # https://github.com/dask/fastparquet/issues/923
+        pytest.skip("fastparquet doesn't work with Numpy 2")
+
     url = "s3://%s/test.parquet" % test_bucket_name
 
     data = pd.DataFrame({"i32": np.array([0, 5, 2, 5])})