Set default ctx in conftest (#1363)

dmlc · Sep 15, 2020 · 8ec0cc7 · 8ec0cc7
1 parent 0e86648
commit 8ec0cc7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/conftest.py b/conftest.py
@@ -228,5 +228,8 @@ def pytest_collection_modifyitems(config, items):
 
 
 def pytest_generate_tests(metafunc):
+    devices = metafunc.config.option.device
+    if not devices:
+        devices = ['cpu']
     if 'ctx' in metafunc.fixturenames:
-        metafunc.parametrize("ctx", [getattr(mx, device)() for device in metafunc.config.option.device])
+        metafunc.parametrize("ctx", [getattr(mx, device)() for device in devices])