tests: update timeouts

AMYPAD · Feb 22, 2024 · 71ef18a · 71ef18a
1 parent 0fea48f
commit 71ef18a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -89,7 +89,7 @@ known_first_party = ["cuvec", "tests"]
 
 [tool.pytest.ini_options]
 minversion = "6.0"
-timeout = 30
+timeout = 5
 log_level = "INFO"
 python_files = ["tests/test_*.py"]
 testpaths = ["tests"]

diff --git a/tests/test_common.py b/tests/test_common.py
@@ -116,6 +116,7 @@ def test_CuVec_creation(cu, classname, caplog):
 
 
 @mark.parametrize("cu", filter(None, [py, sw]))
+@mark.timeout(20)
 def test_asarray(cu):
     v = cu.asarray(np.random.random(shape))
     w = cu.CuVec(v)
@@ -158,6 +159,7 @@ def test_resize(cu):
     assert v._vec.shape == v.shape
 
 
+@mark.timeout(60)
 @mark.parametrize("cu", filter(None, [cp, py, sw]))
 def test_cuda_array_interface(cu):
     cupy = importorskip("cupy")

diff --git a/tests/test_cpython.py b/tests/test_cpython.py
@@ -28,6 +28,7 @@ def test_CVector_strides():
     assert a.strides == (473344, 1376, 4)
 
 
+@mark.timeout(20)
 def test_asarray():
     v = cu.asarray(np.random.random(shape))
     w = cu.CuVec(v)