From 74d52acf3a09d70e92b63263a3e37fe1e729b230 Mon Sep 17 00:00:00 2001 From: fis Date: Sat, 4 Jul 2020 05:14:15 +0800 Subject: [PATCH] Test. --- tests/python-gpu/test_gpu_demos.py | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 tests/python-gpu/test_gpu_demos.py diff --git a/tests/python-gpu/test_gpu_demos.py b/tests/python-gpu/test_gpu_demos.py new file mode 100644 index 000000000000..03c4a627984d --- /dev/null +++ b/tests/python-gpu/test_gpu_demos.py @@ -0,0 +1,11 @@ +import os +import subprocess +import sys +sys.path.append("tests/python") +import test_demos as td # noqa + + +def test_data_iterator(): + script = os.path.join(td.PYTHON_DEMO_DIR, 'data_iterator.py') + cmd = ['python', script] + subprocess.check_call(cmd)