Commit 11ea0449 authored by Victor Stinner's avatar Victor Stinner

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.
parent 2eb6eca3
......@@ -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):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment