Move unit test code into its own file (unders test/ instead of src/).

[dunshire.git] / test / suite.py

diff --git a/test/suite.py b/test/suite.py
index bfd0011ff3b1a40d9e1e82b238766e65e179a8d9..4025647cbe8bb23a1e57e148f48ee8c3fef9bb89 100644 (file)
--- a/test/suite.py
+++ b/test/suite.py
@@ -10,17 +10,19 @@ from doctest import DocTestSuite
 
 # Add '../' to our path.
 from site import addsitedir
-addsitedir('./src/dunshire')
-import cones
-import errors
-import matrices
-import symmetric_linear_game
+addsitedir('./src')
+from dunshire import cones
+from dunshire import errors
+from dunshire import matrices
+from dunshire import games
+import symmetric_linear_game_test
 
 suite = TestSuite()
 suite.addTest(DocTestSuite(cones))
 suite.addTest(DocTestSuite(errors))
 suite.addTest(DocTestSuite(matrices))
-suite.addTest(DocTestSuite(symmetric_linear_game))
-suite.addTest(TestLoader().loadTestsFromModule(symmetric_linear_game))
-runner = TextTestRunner(verbosity=2)
+suite.addTest(DocTestSuite(games))
+suite.addTest(DocTestSuite(symmetric_linear_game_test))
+suite.addTest(TestLoader().loadTestsFromModule(symmetric_linear_game_test))
+runner = TextTestRunner(verbosity=1)
 runner.run(suite)