]> granicus.if.org Git - python/commitdiff
Fix test_inspect messing up with linecache.cache.
authorAntoine Pitrou <solipsis@pitrou.net>
Sun, 8 Jul 2012 11:48:46 +0000 (13:48 +0200)
committerAntoine Pitrou <solipsis@pitrou.net>
Sun, 8 Jul 2012 11:48:46 +0000 (13:48 +0200)
Lib/test/test_inspect.py

index 7ed4ffef8c6430bde3e80bd9da3c1d52f8b95db8..80db03bf0004e4337456ebf2c8e184d8649e983c 100644 (file)
@@ -285,7 +285,10 @@ class TestRetrievingSourceCode(GetSourceBase):
         co = compile("None", fn, "exec")
         self.assertEqual(inspect.getsourcefile(co), None)
         linecache.cache[co.co_filename] = (1, None, "None", co.co_filename)
-        self.assertEqual(normcase(inspect.getsourcefile(co)), fn)
+        try:
+            self.assertEqual(normcase(inspect.getsourcefile(co)), fn)
+        finally:
+            del linecache.cache[co.co_filename]
 
     def test_getfile(self):
         self.assertEqual(inspect.getfile(mod.StupidGit), mod.__file__)
@@ -407,8 +410,11 @@ class TestBuggyCases(GetSourceBase):
         self.assertRaises(IOError, inspect.findsource, co)
         self.assertRaises(IOError, inspect.getsource, co)
         linecache.cache[co.co_filename] = (1, None, lines, co.co_filename)
-        self.assertEqual(inspect.findsource(co), (lines,0))
-        self.assertEqual(inspect.getsource(co), lines[0])
+        try:
+            self.assertEqual(inspect.findsource(co), (lines,0))
+            self.assertEqual(inspect.getsource(co), lines[0])
+        finally:
+            del linecache.cache[co.co_filename]
 
 class TestNoEOL(GetSourceBase):
     def __init__(self, *args, **kwargs):