test: add the ability to run the test suite verbosely.

[dunshire.git] / test / __init__.py

diff --git a/test/__init__.py b/test/__init__.py
index bbcd4ab767f5cdfd4175b3b49a1edbd78f5e7cb7..28d90a6c56098a93e02efabc04ffa7185400c2bf 100644 (file)
--- a/test/__init__.py
+++ b/test/__init__.py
@@ -8,32 +8,46 @@ can be used by setuptools.
 """
 
 from unittest import TestLoader, TestSuite, TextTestRunner
 """
 
 from unittest import TestLoader, TestSuite, TextTestRunner

-from doctest import DocTestSuite
+from doctest import DocTestSuite, ELLIPSIS

 
 from dunshire import cones
 from dunshire import errors
 from dunshire import matrices
 from dunshire import games
 
 from dunshire import cones
 from dunshire import errors
 from dunshire import matrices
 from dunshire import games

+from test import matrices_test
+from test import randomgen

 from test import symmetric_linear_game_test
 
 from test import symmetric_linear_game_test
 

-def build_suite():
+def build_suite(doctests=True):

     """
     Build our test suite, separately from running it.
     """
     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 = TestSuite()

-    suite.addTest(DocTestSuite(cones))
-    suite.addTest(DocTestSuite(errors))
-    suite.addTest(DocTestSuite(matrices))
-    suite.addTest(DocTestSuite(games))
-    suite.addTest(DocTestSuite(symmetric_linear_game_test))
+    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)
     slg_tests = TestLoader().loadTestsFromModule(symmetric_linear_game_test)
     suite.addTest(slg_tests)

+    mat_tests = TestLoader().loadTestsFromModule(matrices_test)
+    suite.addTest(mat_tests)

     return suite
 
     return suite
 

-def run_suite(suite):
+def run_suite(suite, verbosity):

     """
     """

-    Run all of the unit and doctests for the ``dunshire`` and ``test``
-    packages.
+    Run all of the unit and doctests for the :mod:`dunshire` and
+    :mod:`test` packages.

     """
     """

-    runner = TextTestRunner(verbosity=1)
+    runner = TextTestRunner(verbosity=verbosity)

     return runner.run(suite)
     return runner.run(suite)