Enable doctest ELLIPSIS in a few modules.

[dunshire.git] / test / __init__.py

diff --git a/test/__init__.py b/test/__init__.py
index 6942bf2f2af83c8749bae65c586f685ec5b95e08..2188fa60d44401d1293a6e178ba4d7c1225ddda1 100644 (file)
--- a/test/__init__.py
+++ b/test/__init__.py
@@ -8,12 +8,14 @@ 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
 
 def build_suite():
 from test import symmetric_linear_game_test
 
 def build_suite():


@@ -22,18 +24,21 @@ def build_suite():
     """
     suite = TestSuite()
     suite.addTest(DocTestSuite(cones))
     """
     suite = TestSuite()
     suite.addTest(DocTestSuite(cones))

-    suite.addTest(DocTestSuite(errors))
-    suite.addTest(DocTestSuite(matrices))
-    suite.addTest(DocTestSuite(games))
+    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(symmetric_linear_game_test))

+    suite.addTest(DocTestSuite(randomgen))

     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(s):
+def run_suite(suite):

     """
     Run all of the unit and doctests for the ``dunshire`` and ``test``
     packages.
     """
     runner = TextTestRunner(verbosity=1)
     """
     Run all of the unit and doctests for the ``dunshire`` and ``test``
     packages.
     """
     runner = TextTestRunner(verbosity=1)

-    runner.run(s)
+    return runner.run(suite)