From 11ea04491df185c94c12531c6dba6d46fbcb0ad9 Mon Sep 17 00:00:00 2001 From: Victor Stinner Date: Fri, 9 Sep 2016 22:56:54 -0700 Subject: Issue #18401: Fix test_pdb if $HOME is not set HOME is not set on Windows for example. Use also textwrap.dedent() for the script. --- Lib/test/test_pdb.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/Lib/test/test_pdb.py b/Lib/test/test_pdb.py index c39cc97..2076e2f 100644 --- a/Lib/test/test_pdb.py +++ b/Lib/test/test_pdb.py @@ -1057,14 +1057,17 @@ class PdbTestCase(unittest.TestCase): def test_readrc_kwarg(self): - save_home = os.environ['HOME'] + save_home = os.environ.get('HOME', None) save_dir = os.getcwd() - script = """import pdb; pdb.Pdb(readrc=False).set_trace() + script = textwrap.dedent(""" + import pdb; pdb.Pdb(readrc=False).set_trace() -print('hello') -""" - del os.environ['HOME'] + print('hello') + """) try: + if save_home is not None: + del os.environ['HOME'] + with tempfile.TemporaryDirectory() as dirname: os.chdir(dirname) with open('.pdbrc', 'w') as f: @@ -1087,7 +1090,8 @@ print('hello') stdout.decode()) finally: - os.environ['HOME'] = save_home + if save_home is not None: + os.environ['HOME'] = save_home os.chdir(save_dir) def tearDown(self): -- cgit v0.12