Fix broken test and replace redundant generator with a tuple

This commit is contained in:
Nick Coghlan 2012-10-20 00:03:46 +10:00
parent c71b4c7198
commit 62b4136277
1 changed files with 13 additions and 9 deletions

View File

@ -20,16 +20,15 @@
from test import script_helper
def _iter_files(name):
for f in (name + os.extsep + "py",
name + os.extsep + "pyc",
name + os.extsep + "pyo",
name + os.extsep + "pyw",
name + "$py.class"):
yield f
def _files(name):
return (name + os.extsep + "py",
name + os.extsep + "pyc",
name + os.extsep + "pyo",
name + os.extsep + "pyw",
name + "$py.class")
def chmod_files(name):
for f in _iter_files(name):
for f in _files(name):
try:
os.chmod(f, 0o600)
except OSError as exc:
@ -37,7 +36,7 @@ def chmod_files(name):
raise
def remove_files(name):
for f in _iter_files(name):
for f in _files(name):
unlink(f)
rmtree('__pycache__')
@ -160,6 +159,11 @@ def test_rewrite_pyc_with_read_only_source(self):
# Now delete the source file and check the pyc was rewritten
unlink(fname)
unload(TESTFN)
if __debug__:
bytecode_name = fname + "c"
else:
bytecode_name = fname + "o"
os.rename(imp.cache_from_source(fname), bytecode_name)
m3 = __import__(TESTFN)
self.assertEqual(m3.x, 'rewritten')
finally: