mirrors/cpython
0
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2024-11-24 02:15:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Good catch Neal!

Browse Source 
I completely forgot about pyo files and the tests are usually not run with -O. The modified code checks for *.py?
This commit is contained in:
Christian Heimes 2008-01-11 07:03:05 +00:00
parent 61ecb7768f
commit 3540b50884
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Lib/test/test_import.py
View File

@ -213,7 +213,8 @@ class ImportTest(unittest.TestCase):
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)

3
Python/import.c
View File

@ -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);
}