X-Git-Url: https://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=test%2F__init__.py;h=28d90a6c56098a93e02efabc04ffa7185400c2bf;hb=4a9928501f79549742c2e335742ddc65040f744f;hp=ec1bed0b8a5544091c56bbb457059af112f5d338;hpb=040374ca134b2f3d962b91a9dac97a7600032685;p=dunshire.git diff --git a/test/__init__.py b/test/__init__.py index ec1bed0..28d90a6 100644 --- a/test/__init__.py +++ b/test/__init__.py @@ -8,7 +8,7 @@ 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 @@ -18,27 +18,36 @@ 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(games)) - suite.addTest(DocTestSuite(matrices)) - suite.addTest(DocTestSuite(symmetric_linear_game_test)) - suite.addTest(DocTestSuite(randomgen)) + 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 = TextTestRunner(verbosity=verbosity) return runner.run(suite)