Fixed script file to import correctly

Author: Nurdok

pep257

@@ -1,10 +1,6 @@
 #! /usr/bin/env python
-import sys
-from pep257 import main, parse_options
+from pep257 import main
 
 
 if __name__ == '__main__':
-    try:
-        sys.exit(main(*parse_options()))
-    except KeyboardInterrupt:
-        pass
+    main()

pep257.py

@@ -545,7 +545,7 @@ def setup_stream_handler(options):
     log.addHandler(stream_handler)
 
 
-def main():
+def run_pep257():
     opt_parser = get_option_parser()
     # setup the logger before parsing the config file, so that command line
     # arguments for debug / verbose will be printed.
@@ -883,8 +883,12 @@ def SKIP_check_return_type(self, function, docstring):
                 return Error()
 
 
-if __name__ == '__main__':
+def main():
     try:
-        sys.exit(main())
+        sys.exit(run_pep257())
     except KeyboardInterrupt:
         pass
+
+
+if __name__ == '__main__':
+    main()

test_pep257.py

@@ -48,8 +48,8 @@ def open(self, path, *args, **kwargs):
 
     def invoke_pep257(self, args=""):
         """Run pep257.py on the environment base folder with the given args."""
-        pep257_location = os.path.join(os.path.dirname(__file__), 'pep257.py')
-        cmd = shlex.split("python {0} {1} {2}"
+        pep257_location = os.path.join(os.path.dirname(__file__), 'pep257')
+        cmd = shlex.split("{0} {1} {2}"
                           .format(pep257_location, self.tempdir, args))
         p = subprocess.Popen(cmd, stdout=subprocess.PIPE,
                              stderr=subprocess.PIPE)