Test dask demos. (#8557)

Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>

tests/test_distributed/test_gpu_with_dask/test_gpu_demos.py

@@ -0,0 +1,25 @@
+import os
+import subprocess
+
+import pytest
+
+from xgboost import testing as tm
+
+
+@pytest.mark.skipif(**tm.no_dask())
+@pytest.mark.skipif(**tm.no_dask_cuda())
+@pytest.mark.skipif(**tm.no_cupy())
+@pytest.mark.mgpu
+def test_dask_training():
+    script = os.path.join(tm.demo_dir(__file__), "dask", "gpu_training.py")
+    cmd = ["python", script]
+    subprocess.check_call(cmd)
+
+
+@pytest.mark.skipif(**tm.no_dask_cuda())
+@pytest.mark.skipif(**tm.no_dask())
+@pytest.mark.mgpu
+def test_dask_sklearn_demo():
+    script = os.path.join(tm.demo_dir(__file__), "dask", "sklearn_gpu_training.py")
+    cmd = ["python", script]
+    subprocess.check_call(cmd)

tests/test_distributed/test_with_dask/test_demos.py

@@ -0,0 +1,36 @@
+import os
+import subprocess
+
+import pytest
+
+from xgboost import testing as tm
+
+
+@pytest.mark.skipif(**tm.no_dask())
+def test_dask_cpu_training_demo():
+    script = os.path.join(tm.demo_dir(__file__), "dask", "cpu_training.py")
+    cmd = ["python", script]
+    subprocess.check_call(cmd)
+
+
+@pytest.mark.skipif(**tm.no_dask())
+def test_dask_cpu_survival_demo():
+    script = os.path.join(tm.demo_dir(__file__), "dask", "cpu_survival.py")
+    cmd = ["python", script]
+    subprocess.check_call(cmd)
+
+
+# Not actually run on CI due to missing dask_ml.
+@pytest.mark.skipif(**tm.no_dask())
+@pytest.mark.skipif(**tm.no_dask_ml())
+def test_dask_callbacks_demo():
+    script = os.path.join(tm.demo_dir(__file__), "dask", "dask_callbacks.py")
+    cmd = ["python", script]
+    subprocess.check_call(cmd)
+
+
+@pytest.mark.skipif(**tm.no_dask())
+def test_dask_sklearn_demo():
+    script = os.path.join(tm.demo_dir(__file__), "dask", "sklearn_cpu_training.py")
+    cmd = ["python", script]
+    subprocess.check_call(cmd)