X-Git-Url: https://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=test%2F__init__.py;h=28d90a6c56098a93e02efabc04ffa7185400c2bf;hb=4a9928501f79549742c2e335742ddc65040f744f;hp=4241453f5a99fddb29d3e682244bca48a8f48799;hpb=59a1dcf2bf416a2527f9fdfb377afbbfa6cef696;p=dunshire.git

diff --git a/test/__init__.py b/test/__init__.py
index 4241453..28d90a6 100644
--- a/test/__init__.py
+++ b/test/__init__.py
@@ -8,32 +8,46 @@ can be used by setuptools.
 """
 
 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 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))
-    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)
+    mat_tests = TestLoader().loadTestsFromModule(matrices_test)
+    suite.addTest(mat_tests)
     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.run(suite)
+    runner = TextTestRunner(verbosity=verbosity)
+    return runner.run(suite)