From f71f94d6aa7b962a715652d2fc9fcdba3a35b97c Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Thu, 7 May 2009 21:19:40 -0400 Subject: Treat files not ending in .py properly. --- coverage/control.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'coverage/control.py') diff --git a/coverage/control.py b/coverage/control.py index 3c6b55a..697e732 100644 --- a/coverage/control.py +++ b/coverage/control.py @@ -40,7 +40,6 @@ class coverage: should not. """ - ofilename = filename if filename == '': # There's no point in ever tracing string executions, we can't do # anything with the data later anyway. @@ -55,13 +54,12 @@ class coverage: dunder_file = frame.f_globals.get('__file__') if dunder_file: if not dunder_file.endswith(".py"): - dunder_file = dunder_file[:-1] + if dunder_file[-4:-1] == ".py": + dunder_file = dunder_file[:-1] filename = dunder_file canonical = self.file_locator.canonical_filename(filename) - #print "of: %r\nfn: %r\n__: %r\nca: %r\n" % (ofilename, filename, dunder_file, canonical) - # If we aren't supposed to trace the stdlib, then check if this is in # the stdlib and skip it if so. if not self.cover_stdlib: -- cgit v1.2.1