From: Georg Brandl <georg@python.org>
Date: Tue, 1 May 2012 07:21:16 +0000 (+0200)
Subject: test_pdb: fix failure of test_issue13183 in debug mode, and make sure files are clean... 
X-Git-Tag: v3.3.0a3~4
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=4bde9caf7476eb9a9fc360eda2918962d954720b;p=python

test_pdb: fix failure of test_issue13183 in debug mode, and make sure files are cleaned up.
---

diff --git a/Lib/test/test_pdb.py b/Lib/test/test_pdb.py
index ac929312e3..157087351e 100644
--- a/Lib/test/test_pdb.py
+++ b/Lib/test/test_pdb.py
@@ -604,6 +604,7 @@ class PdbTestCase(unittest.TestCase):
         filename = 'main.py'
         with open(filename, 'w') as f:
             f.write(textwrap.dedent(script))
+        self.addCleanup(support.unlink, filename)
         cmd = [sys.executable, '-m', 'pdb', filename]
         stdout = stderr = None
         with subprocess.Popen(cmd, stdout=subprocess.PIPE,
@@ -660,9 +661,11 @@ class PdbTestCase(unittest.TestCase):
         """
         with open('bar.py', 'w') as f:
             f.write(textwrap.dedent(bar))
+        self.addCleanup(support.unlink, 'bar.py')
         stdout, stderr = self.run_pdb(script, commands)
-        self.assertIn('main.py(5)foo()->None', stdout.split('\n')[-3],
-                         'Fail to step into the caller after a return')
+        self.assertTrue(
+            any('main.py(5)foo()->None' in l for l in stdout.splitlines()),
+            'Fail to step into the caller after a return')
 
     def tearDown(self):
         support.unlink(support.TESTFN)