Make invoke command discover and run doctests in the scrapi module

NeuroVault · Sep 15, 2015 · 4e03402 · 4e03402
1 parent b02c3a0
commit 4e03402
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tasks.py b/tasks.py
@@ -112,11 +112,13 @@ def elasticsearch():
 
 
 @task
-def test(cov=True, verbose=False, debug=False):
+def test(cov=True, verbose=False, debug=False, doctests=True):
     """
     Runs all tests in the 'tests/' directory
     """
-    cmd = 'py.test tests'
+    cmd = 'py.test scrapi tests'
+    if doctests:
+        cmd += ' --doctest-modules'
     if verbose:
         cmd += ' -v'
     if debug: