Added timeouts to benchmark tests

conda/environments/all_cuda-118_arch-aarch64.yaml

@@ -29,6 +29,7 @@ dependencies:
 - pre-commit
 - pytest
 - pytest-cov
+- pytest-timeout
 - python>=3.10,<3.13
 - rangehttpserver
 - rapids-build-backend>=0.3.0,<0.4.0.dev0

conda/environments/all_cuda-118_arch-x86_64.yaml

@@ -31,6 +31,7 @@ dependencies:
 - pre-commit
 - pytest
 - pytest-cov
+- pytest-timeout
 - python>=3.10,<3.13
 - rangehttpserver
 - rapids-build-backend>=0.3.0,<0.4.0.dev0

conda/environments/all_cuda-128_arch-aarch64.yaml

@@ -29,6 +29,7 @@ dependencies:
 - pre-commit
 - pytest
 - pytest-cov
+- pytest-timeout
 - python>=3.10,<3.13
 - rangehttpserver
 - rapids-build-backend>=0.3.0,<0.4.0.dev0

conda/environments/all_cuda-128_arch-x86_64.yaml

@@ -29,6 +29,7 @@ dependencies:
 - pre-commit
 - pytest
 - pytest-cov
+- pytest-timeout
 - python>=3.10,<3.13
 - rangehttpserver
 - rapids-build-backend>=0.3.0,<0.4.0.dev0

dependencies.yaml

@@ -420,6 +420,7 @@ dependencies:
           - rapids-dask-dependency==25.2.*,>=0.0.0a0
           - pytest
           - pytest-cov
+          - pytest-timeout
           - rangehttpserver
           - boto3>=1.21.21
       - output_types: [requirements, pyproject]

python/kvikio/pyproject.toml

@@ -45,6 +45,7 @@ test = [
     "moto[server]>=4.0.8",
     "pytest",
     "pytest-cov",
+    "pytest-timeout",
     "rangehttpserver",
     "rapids-dask-dependency==25.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.

python/kvikio/tests/test_benchmarks.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2022-2024, NVIDIA CORPORATION. All rights reserved.
+# Copyright (c) 2022-2025, NVIDIA CORPORATION. All rights reserved.
 # See file LICENSE for terms.
 
 import os
@@ -28,6 +28,7 @@
         "zarr",
     ],
 )
+@pytest.mark.timeout(30)
 def test_single_node_io(run_cmd, tmp_path, api):
     """Test benchmarks/single_node_io.py"""
 
@@ -59,6 +60,7 @@ def test_single_node_io(run_cmd, tmp_path, api):
         "posix",
     ],
 )
+@pytest.mark.timeout(30)
 def test_zarr_io(run_cmd, tmp_path, api):
     """Test benchmarks/zarr_io.py"""
 
@@ -89,6 +91,7 @@ def test_zarr_io(run_cmd, tmp_path, api):
         "numpy",
     ],
 )
+@pytest.mark.timeout(30)
 def test_http_io(run_cmd, api):
     """Test benchmarks/http_io.py"""
 
@@ -118,6 +121,7 @@ def test_http_io(run_cmd, api):
         "numpy",
     ],
 )
+@pytest.mark.timeout(30)
 def test_s3_io(run_cmd, api):
     """Test benchmarks/s3_io.py"""