Good catch Neal!

tiran · tiran · commit 3540b5088443 · 2008-01-11T07:03:05.000Z
I completely forgot about pyo files and the tests are usually not run with -O. The modified code checks for *.py?
diff --git a/Lib/test/test_import.py b/Lib/test/test_import.py
@@ -213,7 +213,8 @@ def test_file_to_source(self):
             os.remove(source)
             del sys.modules[TESTFN]
             mod = __import__(TESTFN)
-            self.failUnless(mod.__file__.endswith('.pyc'))
+            ext = mod.__file__[-4:]
+            self.failUnless(ext in ('.pyc', '.pyo'), ext)
         finally:
             sys.path.pop(0)
             remove_files(TESTFN)
diff --git a/Python/import.c b/Python/import.c
@@ -982,7 +982,8 @@ get_sourcefile(const char *file)
 	}
 
 	len = strlen(file);
-	if (len > MAXPATHLEN || PyOS_stricmp(&file[len-4], ".pyc") != 0) {
+        /* match '*.py?' */
+	if (len > MAXPATHLEN || PyOS_strnicmp(&file[len-4], ".py", 3) != 0) {
 		return PyUnicode_DecodeFSDefault(file);
 	}
 

Original file line number	Diff line number	Diff line change
`@@ -982,7 +982,8 @@ get_sourcefile(const char *file)`
`982`	`982`	`}`
`983`	`983`
`984`	`984`	`len = strlen(file);`
`985`		`- if (len > MAXPATHLEN \|\| PyOS_stricmp(&file[len-4], ".pyc") != 0) {`
	`985`	`+ /* match '.py?' /`
	`986`	`+ if (len > MAXPATHLEN \|\| PyOS_strnicmp(&file[len-4], ".py", 3) != 0) {`
`986`	`987`	`return PyUnicode_DecodeFSDefault(file);`
`987`	`988`	`}`
`988`	`989`