diff options
author | Victor Stinner <victor.stinner@gmail.com> | 2016-09-09 22:56:54 -0700 |
---|---|---|
committer | Victor Stinner <victor.stinner@gmail.com> | 2016-09-09 22:56:54 -0700 |
commit | 7223c72a329aeac30466409f9453cb676abcffdf (patch) | |
tree | 0a334c561a06fe362ce41ad0c4cc73bc0a9cac8f /Lib/test/test_pdb.py | |
parent | a67d22124d294b6b165eda9fd65bb08997ab6f53 (diff) | |
download | cpython-7223c72a329aeac30466409f9453cb676abcffdf.tar.gz |
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.
Diffstat (limited to 'Lib/test/test_pdb.py')
-rw-r--r-- | Lib/test/test_pdb.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/Lib/test/test_pdb.py b/Lib/test/test_pdb.py index c39cc97fe1..2076e2f3e0 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): |