Add more doctest examples to the randomgen module.

[dunshire.git] / test / __init__.py

diff --git a/test/__init__.py b/test/__init__.py
index 1d7c9a907394483b2542fa3253b2d24c58e8a9b2..0f6f15c665da78f61effd04e9159e5bf50b9ef56 100644 (file)
--- a/test/__init__.py
+++ b/test/__init__.py
@@ -8,12 +8,13 @@ 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 randomgen
 from test import symmetric_linear_game_test
 


@@ -23,13 +24,15 @@ def build_suite():
     """
     suite = TestSuite()
     suite.addTest(DocTestSuite(cones))
     """
     suite = TestSuite()
     suite.addTest(DocTestSuite(cones))

-    suite.addTest(DocTestSuite(errors))
-    suite.addTest(DocTestSuite(games))
-    suite.addTest(DocTestSuite(matrices))
+    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))
+    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
 
 def run_suite(suite):
     return suite
 
 def run_suite(suite):