Another patch for #1762972: __file__ points to the py file instead pyo/pyc file

This commit is contained in:
Christian Heimes 2008-01-07 20:12:44 +00:00
parent a38f73b1bb
commit 3b06e5378a
3 changed files with 63 additions and 5 deletions

View file

@ -1,4 +1,4 @@
from test.test_support import TESTFN, run_unittest, catch_warning
from test.test_support import TESTFN, run_unittest, catch_warning
import unittest
import os
@ -200,6 +200,27 @@ class ImportTest(unittest.TestCase):
if TESTFN in sys.modules:
del sys.modules[TESTFN]
def test_file_to_source(self):
# check if __file__ points to the source file where available
source = TESTFN + ".py"
with open(source, "w") as f:
f.write("test = None\n")
sys.path.insert(0, os.curdir)
try:
mod = __import__(TESTFN)
self.failUnless(mod.__file__.endswith('.py'))
os.remove(source)
del sys.modules[TESTFN]
mod = __import__(TESTFN)
self.failUnless(mod.__file__.endswith('.pyc'))
finally:
sys.path.pop(0)
remove_files(TESTFN)
if TESTFN in sys.modules:
del sys.modules[TESTFN]
class PathsTests(unittest.TestCase):
SAMPLES = ('test', 'test\u00e4\u00f6\u00fc\u00df', 'test\u00e9\u00e8',
'test\u00b0\u00b3\u00b2')