]> granicus.if.org Git - python/commitdiff
Merged revisions 80194 via svnmerge from
authorAntoine Pitrou <solipsis@pitrou.net>
Sun, 18 Apr 2010 19:17:01 +0000 (19:17 +0000)
committerAntoine Pitrou <solipsis@pitrou.net>
Sun, 18 Apr 2010 19:17:01 +0000 (19:17 +0000)
svn+ssh://pythondev@svn.python.org/python/trunk

........
  r80194 | antoine.pitrou | 2010-04-18 21:14:38 +0200 (dim., 18 avril 2010) | 3 lines

  Fix catastrophic file opening and closing logic in test_linecache
........

Lib/test/test_linecache.py

index 3715c8a1b06249c976e94d7fcd1c1848956b9197..994125037882e83900f591988dd46c8f3951b47d 100644 (file)
@@ -86,31 +86,28 @@ class LineCacheTests(unittest.TestCase):
             source_name = support.TESTFN + '.py'
             with open(source_name, 'w') as source:
                 source.write(SOURCE_1)
-                source.close()
-                getline(source_name, 1)
+            getline(source_name, 1)
 
-                # Keep a copy of the old contents
-                source_list = []
-                source = open(source_name)
+            # Keep a copy of the old contents
+            source_list = []
+            with open(source_name) as source:
                 for index, line in enumerate(source):
                     self.assertEquals(line, getline(source_name, index + 1))
                     source_list.append(line)
-                source.close()
 
-                source = open(source_name, 'w')
+            with open(source_name, 'w') as source:
                 source.write(SOURCE_2)
-                source.close()
 
-                # Try to update a bogus cache entry
-                linecache.checkcache('dummy')
+            # Try to update a bogus cache entry
+            linecache.checkcache('dummy')
 
-                # Check that the cache matches the old contents
-                for index, line in enumerate(source_list):
-                    self.assertEquals(line, getline(source_name, index + 1))
+            # Check that the cache matches the old contents
+            for index, line in enumerate(source_list):
+                self.assertEquals(line, getline(source_name, index + 1))
 
-                # Update the cache and check whether it matches the new source file
-                linecache.checkcache(source_name)
-                source = open(source_name)
+            # Update the cache and check whether it matches the new source file
+            linecache.checkcache(source_name)
+            with open(source_name) as source:
                 for index, line in enumerate(source):
                     self.assertEquals(line, getline(source_name, index + 1))
                     source_list.append(line)