Multi with statement

Lib/test/test_compiler.py

 import test.test_support
 compiler = test.test_support.import_module('compiler', deprecated=True)
 from compiler.ast import flatten
 import os, sys, time, unittest
 from random import random
 from StringIO import StringIO
 
 # How much time in seconds can pass before we print a 'Still working' message.
 _PRINT_WORKING_MSG_INTERVAL = 5 * 60
 
@@ skipping 147 matching lines @@
                              'def f():\n'
                              '    with TrivialContext() as tc:\n'
                              '        return 1\n'
                              'result = f()',
                              '<string>',
                              'exec' )
         dct = {'TrivialContext': TrivialContext}
         exec c in dct
         self.assertEquals(dct.get('result'), 1)
 
+    def testWithMult(self):
+        events = []
+        class Ctx:
+            def __init__(self, n):
+                self.n = n
+            def __enter__(self):
+                events.append(self.n)
+            def __exit__(self, *args):
+                pass
+        c = compiler.compile('from __future__ import with_statement\n'
+                             'def f():\n'
+                             '    with Ctx(1) as tc, Ctx(2) as tc2:\n'
+                             '        return 1\n'
+                             'result = f()',
+                             '<string>',
+                             'exec' )
+        dct = {'Ctx': Ctx}
+        exec c in dct
+        self.assertEquals(dct.get('result'), 1)

[Benjamin, 2009/05/02 18:22:06]

Why using .get() here?

[Georg, 2009/05/02 18:48:00]

Following the above example.  Probably to not get an error but a test failure if
it isn't there.

+        self.assertEquals(events, [1, 2])
+
     def testGlobal(self):
         code = compiler.compile('global x\nx=1', '<string>', 'exec')
         d1 = {'__builtins__': {}}
         d2 = {}
         exec code in d1, d2
         # x should be in the globals dict
         self.assertEquals(d1.get('x'), 1)
 
     def testPrintFunction(self):
         c = compiler.compile('from __future__ import print_function\n'
@@ skipping 74 matching lines @@
 
 ###############################################################################
 
 def test_main():
     global TEST_ALL
     TEST_ALL = test.test_support.is_resource_enabled("compiler")
     test.test_support.run_unittest(CompilerTest)
 
 if __name__ == "__main__":
     test_main()