Kaydet (Commit) 9958c56e authored tarafından Antoine Pitrou's avatar Antoine Pitrou

Fix catastrophic file opening and closing logic in test_linecache

üst 814b6c22
...@@ -86,31 +86,28 @@ class LineCacheTests(unittest.TestCase): ...@@ -86,31 +86,28 @@ class LineCacheTests(unittest.TestCase):
source_name = support.TESTFN + '.py' source_name = support.TESTFN + '.py'
with open(source_name, 'w') as source: with open(source_name, 'w') as source:
source.write(SOURCE_1) source.write(SOURCE_1)
source.close() getline(source_name, 1)
getline(source_name, 1)
# Keep a copy of the old contents # Keep a copy of the old contents
source_list = [] source_list = []
source = open(source_name) with open(source_name) as source:
for index, line in enumerate(source): for index, line in enumerate(source):
self.assertEquals(line, getline(source_name, index + 1)) self.assertEquals(line, getline(source_name, index + 1))
source_list.append(line) source_list.append(line)
source.close()
source = open(source_name, 'w') with open(source_name, 'w') as source:
source.write(SOURCE_2) source.write(SOURCE_2)
source.close()
# Try to update a bogus cache entry # Try to update a bogus cache entry
linecache.checkcache('dummy') linecache.checkcache('dummy')
# Check that the cache matches the old contents # Check that the cache matches the old contents
for index, line in enumerate(source_list): for index, line in enumerate(source_list):
self.assertEquals(line, getline(source_name, index + 1)) self.assertEquals(line, getline(source_name, index + 1))
# Update the cache and check whether it matches the new source file # Update the cache and check whether it matches the new source file
linecache.checkcache(source_name) linecache.checkcache(source_name)
source = open(source_name) with open(source_name) as source:
for index, line in enumerate(source): for index, line in enumerate(source):
self.assertEquals(line, getline(source_name, index + 1)) self.assertEquals(line, getline(source_name, index + 1))
source_list.append(line) source_list.append(line)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment