Pylint cleanup for tests.

[dunshire.git] / src / test / suite.py

diff --git a/src/test/suite.py b/src/test/suite.py
index 078d0c9a3128680cfdbf5b16d41a9133d929db56..e10c38d98a3401a2f0a37fa033f77d3e2270863a 100644 (file)
--- a/src/test/suite.py
+++ b/src/test/suite.py
@@ -8,22 +8,27 @@ codebase into one big TestSuite() and the runs it.
 from unittest import TestLoader, TestSuite, TextTestRunner
 from doctest import DocTestSuite
 
-# Add '../' to our path.
-from site import addsitedir
-addsitedir('./src')
 from dunshire import cones
 from dunshire import errors
 from dunshire import matrices
 from dunshire import games
 from test import symmetric_linear_game_test
 
-if __name__ == '__main__':
+def run_suite():
+    """
+    Run all of the unit and doctests for the ``dunshire`` and ``test``
+    packages.
+    """
     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))
-    suite.addTest(TestLoader().loadTestsFromModule(symmetric_linear_game_test))
+    slg_tests = TestLoader().loadTestsFromModule(symmetric_linear_game_test)
+    suite.addTest(slg_tests)
     runner = TextTestRunner(verbosity=1)
     runner.run(suite)
+
+if __name__ == '__main__':
+    run_suite()