Check multiple different timeout error messages

rapidsai · Aug 23, 2023 · 396688f · 396688f
1 parent f022c3d
commit 396688f
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/dask_cuda/tests/test_dask_cuda_worker.py b/dask_cuda/tests/test_dask_cuda_worker.py
@@ -450,6 +450,12 @@ def test_worker_timeout():
     )
 
     assert "closing nanny at" in ret.stderr.lower()
-    assert "reason: failure-to-start-" in ret.stderr.lower()
-    assert "timeouterror" in ret.stderr.lower()
+
+    # Depending on the environment, the error raised may be different
+    try:
+        assert "reason: failure-to-start-" in ret.stderr.lower()
+        assert "timeouterror" in ret.stderr.lower()
+    except AssertionError:
+        assert "reason: nanny-close" in ret.stderr.lower()
+
     assert ret.returncode == 0