# 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
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))
+suite.addTest(DocTestSuite(games))
+suite.addTest(TestLoader().loadTestsFromModule(games))
runner = TextTestRunner(verbosity=2)
runner.run(suite)