#13183: backport fixes to test_pdb to 2.7 branch
diff --git a/Lib/test/test_pdb.py b/Lib/test/test_pdb.py
index 5c1a37c..559f756 100644
--- a/Lib/test/test_pdb.py
+++ b/Lib/test/test_pdb.py
@@ -20,6 +20,7 @@
filename = 'main.py'
with open(filename, 'w') as f:
f.write(textwrap.dedent(script))
+ self.addCleanup(test_support.unlink, filename)
cmd = [sys.executable, '-m', 'pdb', filename]
stdout = stderr = None
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE,
@@ -61,9 +62,11 @@
"""
with open('bar.py', 'w') as f:
f.write(textwrap.dedent(bar))
+ self.addCleanup(test_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')
class PdbTestInput(object):