diff --git a/Lib/test/test_pdb.py b/Lib/test/test_pdb.py index 5c1a37c280a..559f75623f4 100644 --- a/Lib/test/test_pdb.py +++ b/Lib/test/test_pdb.py @@ -20,6 +20,7 @@ class PdbTestCase(unittest.TestCase): 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 @@ class PdbTestCase(unittest.TestCase): """ 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):