diff --git a/ci/test_python.sh b/ci/test_python.sh index 73a93fcac..827eb84c9 100755 --- a/ci/test_python.sh +++ b/ci/test_python.sh @@ -47,6 +47,7 @@ UCX_WARN_UNUSED_ENV_VARS=n \ UCX_MEMTYPE_CACHE=n \ timeout 40m pytest \ -vv \ + --durations=0 \ --capture=no \ --cache-clear \ --junitxml="${RAPIDS_TESTS_DIR}/junit-dask-cuda.xml" \ diff --git a/dask_cuda/tests/test_local_cuda_cluster.py b/dask_cuda/tests/test_local_cuda_cluster.py index 530e51e2d..845759dfd 100644 --- a/dask_cuda/tests/test_local_cuda_cluster.py +++ b/dask_cuda/tests/test_local_cuda_cluster.py @@ -454,6 +454,7 @@ async def test_get_cluster_configuration(): @gen_test(timeout=20) async def test_worker_fraction_limits(): async with LocalCUDACluster( + dashboard_address=None, device_memory_limit=0.1, rmm_pool_size=0.2, rmm_maximum_pool_size=0.3, diff --git a/dask_cuda/tests/test_proxify_host_file.py b/dask_cuda/tests/test_proxify_host_file.py index 50b2c51a5..2e3f8269d 100644 --- a/dask_cuda/tests/test_proxify_host_file.py +++ b/dask_cuda/tests/test_proxify_host_file.py @@ -384,7 +384,7 @@ def test_incompatible_types(root_dir): @pytest.mark.parametrize("npartitions", [1, 2, 3]) @pytest.mark.parametrize("compatibility_mode", [True, False]) -@gen_test(timeout=20) +@gen_test(timeout=30) async def test_compatibility_mode_dataframe_shuffle(compatibility_mode, npartitions): cudf = pytest.importorskip("cudf") diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py index cfdbf636b..53282bef1 100644 --- a/dask_cuda/tests/test_proxy.py +++ b/dask_cuda/tests/test_proxy.py @@ -400,7 +400,7 @@ def _pxy_deserialize(self): @pytest.mark.parametrize("send_serializers", [None, ("dask", "pickle"), ("cuda",)]) @pytest.mark.parametrize("protocol", ["tcp", "ucx"]) -@gen_test(timeout=20) +@gen_test(timeout=60) async def test_communicating_proxy_objects(protocol, send_serializers): """Testing serialization of cuDF dataframe when communicating""" cudf = pytest.importorskip("cudf")