Build looping into the test suite and make doctests optional.

[dunshire.git] / test / __init__.py

diff --git a/test/__init__.py b/test/__init__.py
index 0f6f15c665da78f61effd04e9159e5bf50b9ef56..2e23a7c7dbec733dc994b611e137a33f419d036d 100644 (file)
--- a/test/__init__.py
+++ b/test/__init__.py
@@ -18,17 +18,26 @@ from test import matrices_test
 from test import randomgen
 from test import symmetric_linear_game_test
 
-def build_suite():
+def build_suite(doctests=True):
     """
     Build our test suite, separately from running it.
+
+    Parameters
+    ----------
+
+    doctests : bool
+        Do you want to build the doctests, too? During random testing,
+        the answer may be "no."
+
     """
     suite = TestSuite()
-    suite.addTest(DocTestSuite(cones))
-    suite.addTest(DocTestSuite(errors, optionflags=ELLIPSIS))
-    suite.addTest(DocTestSuite(games, optionflags=ELLIPSIS))
-    suite.addTest(DocTestSuite(matrices, optionflags=ELLIPSIS))
-    suite.addTest(DocTestSuite(symmetric_linear_game_test))
-    suite.addTest(DocTestSuite(randomgen, optionflags=ELLIPSIS))
+    if doctests:
+        suite.addTest(DocTestSuite(cones))
+        suite.addTest(DocTestSuite(errors, optionflags=ELLIPSIS))
+        suite.addTest(DocTestSuite(games, optionflags=ELLIPSIS))
+        suite.addTest(DocTestSuite(matrices, optionflags=ELLIPSIS))
+        suite.addTest(DocTestSuite(symmetric_linear_game_test))
+        suite.addTest(DocTestSuite(randomgen, optionflags=ELLIPSIS))
     slg_tests = TestLoader().loadTestsFromModule(symmetric_linear_game_test)
     suite.addTest(slg_tests)
     mat_tests = TestLoader().loadTestsFromModule(matrices_test)