Test scripts now work with and without source.

1 files changed, 10 insertions, 1 deletions

diff --git a/test_pystache.py b/test_pystache.py
index 0f997e5..9a1a3ca 100644
--- a/test_pystache.py
+++ b/test_pystache.py
@@ -15,7 +15,16 @@ in Python 2.4:
 
 """
 
-from pystache.commands.test import main
+import sys
+
+from pystache.commands import test
+from pystache.tests.main import FROM_SOURCE_OPTION
+
+
+def main(sys_argv=sys.argv):
+    sys.argv.insert(1, FROM_SOURCE_OPTION)
+    test.main()
+
 
 if __name__=='__main__':
     main()