Python source files that don't end with a newline can now be executed, fixing #47.

1 files changed, 8 insertions, 2 deletions

diff --git a/coverage/execfile.py b/coverage/execfile.py
index f04840b..1a2ffad 100644
--- a/coverage/execfile.py
+++ b/coverage/execfile.py
@@ -2,7 +2,7 @@
 
 import imp, os, sys
 
-from coverage.backward import exec_function
+from coverage.backward import exec_code_object
 from coverage.misc import NoSource, ExceptionDuringRun
 
 
@@ -42,9 +42,15 @@ def run_python_file(filename, args):
         except IOError:
             raise NoSource("No file to run: %r" % filename)
 
+        # We have the source.  `compile` still needs the last line to be clean,
+        # so make sure it is, then compile a code object from it.
+        if source[-1] != '\n':
+            source += '\n'
+        code = compile(source, filename, "exec")
+
         # Execute the source file.
         try:
-            exec_function(source, filename, main_mod.__dict__)
+            exec_code_object(code, main_mod.__dict__)
         except:
             # Something went wrong while executing the user code.
             # Get the exc_info, and pack them into an exception that we can