X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2Ftest%2Fsuite.py;fp=src%2Ftest%2Fsuite.py;h=e10c38d98a3401a2f0a37fa033f77d3e2270863a;hb=dd7758c84633a2c72b164e440078ab737abea701;hp=078d0c9a3128680cfdbf5b16d41a9133d929db56;hpb=a050ae7de2580d78ddb157262560542e7dc3abac;p=dunshire.git

diff --git a/src/test/suite.py b/src/test/suite.py
index 078d0c9..e10c38d 100644
--- 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()