changeset:   103553:09c730db1aac
user:        Victor Stinner <victor.stinner@gmail.com>
date:        Fri Sep 09 22:56:54 2016 -0700
files:       Lib/test/test_pdb.py
description:
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.


diff -r ee0bbfd972de -r 09c730db1aac Lib/test/test_pdb.py
--- a/Lib/test/test_pdb.py	Fri Sep 09 22:21:17 2016 -0700
+++ b/Lib/test/test_pdb.py	Fri Sep 09 22:56:54 2016 -0700
@@ -1057,14 +1057,17 @@
 
 
     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 @@
                               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):