Updated pystache/commands/test.py to use the new test-discovery code.

1 files changed, 5 insertions, 11 deletions

diff --git a/pystache/commands/test.py b/pystache/commands/test.py
index 8588002..4bf0a4b 100644
--- a/pystache/commands/test.py
+++ b/pystache/commands/test.py
@@ -7,20 +7,14 @@ This module provides a command to test pystache (unit tests, doctests, etc).
 
 import sys
 
-# TODO: make nose unnecessary.
-import nose
-
 import pystache
+from pystache.tests.main import Tester
 
 
-def main(sys_argv):
-    # This does not work with the --with-doctest flag yet because of the
-    # following issue:
-    #   https://github.com/nose-devs/nose/issues/383
-    # TODO: change module keyword argument to defaultTest keyword argument:
-    #   http://readthedocs.org/docs/nose/en/latest/api/core.html#module-nose.core
-    nose.main(module=pystache)
+def main(sys_argv=sys.argv):
+    tester = Tester()
+    tester.run_tests(package=pystache, sys_argv=sys_argv)
 
 
 if __name__=='__main__':
-    main(sys.argv)
+    main()