Add test for unknown CLI argument

rapidsai · Jan 4, 2021 · 6e144ff · 6e144ff
1 parent 84f86b7
commit 6e144ff
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/dask_cuda/tests/test_dask_cuda_worker.py b/dask_cuda/tests/test_dask_cuda_worker.py
@@ -1,6 +1,7 @@
 from __future__ import absolute_import, division, print_function
 
 import os
+import subprocess
 
 import pytest
 
@@ -98,3 +99,9 @@ def test_rmm_managed(loop):  # noqa: F811
                 )
                 for v in memory_resource_type.values():
                     assert v is rmm.mr.ManagedMemoryResource
+
+
+def test_unknown_argument():
+    ret = subprocess.run(["dask-cuda-worker", "--my-argument"], capture_output=True)
+    assert ret.returncode != 0
+    assert b"Scheduler address: --my-argument" in ret.stderr