diff options
Diffstat (limited to 'Lib/test/test_linecache.py')
| -rw-r--r-- | Lib/test/test_linecache.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/Lib/test/test_linecache.py b/Lib/test/test_linecache.py index b5f80a4f43e..7d14534d35b 100644 --- a/Lib/test/test_linecache.py +++ b/Lib/test/test_linecache.py @@ -3,7 +3,7 @@ import linecache import unittest import os.path -from test import test_support as support +from test import support FILENAME = linecache.__file__ @@ -55,14 +55,16 @@ class LineCacheTests(unittest.TestCase): # Check whether lines correspond to those from file iteration for entry in TESTS: filename = os.path.join(TEST_PATH, entry) + '.py' - for index, line in enumerate(open(filename)): - self.assertEqual(line, getline(filename, index + 1)) + with open(filename) as file: + for index, line in enumerate(file): + self.assertEqual(line, getline(filename, index + 1)) # Check module loading for entry in MODULES: filename = os.path.join(MODULE_PATH, entry) + '.py' - for index, line in enumerate(open(filename)): - self.assertEqual(line, getline(filename, index + 1)) + with open(filename) as file: + for index, line in enumerate(file): + self.assertEqual(line, getline(filename, index + 1)) # Check that bogus data isn't returned (issue #1309567) empty = linecache.getlines('a/b/c/__init__.py') |