[pypy-svn] r40423 - pypy/dist/pypy/translator/goal/test

[arigo at codespeak.net]

Author: arigo
Date: Tue Mar 13 11:13:39 2007
New Revision: 40423

Modified:
   pypy/dist/pypy/translator/goal/test/test_app_main.py
Log:
Don't crash if $PYTHONSTARTUP is not set in the outside shell.


Modified: pypy/dist/pypy/translator/goal/test/test_app_main.py
==============================================================================
--- pypy/dist/pypy/translator/goal/test/test_app_main.py	(original)
+++ pypy/dist/pypy/translator/goal/test/test_app_main.py	Tue Mar 13 11:13:39 2007
@@ -184,7 +184,7 @@
         child.expect('foobye')
 
     def test_pythonstartup(self):
-        old = os.environ['PYTHONSTARTUP']
+        old = os.environ.get('PYTHONSTARTUP', '')
         try:
             os.environ['PYTHONSTARTUP'] = crashing_demo_script
             child = self.spawn([])
@@ -278,7 +278,7 @@
         assert '7776' in data
 
     def test_no_pythonstartup(self):
-        old = os.environ['PYTHONSTARTUP']
+        old = os.environ.get('PYTHONSTARTUP', '')
         try:
             os.environ['PYTHONSTARTUP'] = crashing_demo_script
             data = self.run('"%s"' % (demo_script,))