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.

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):