Lib/test/test_transformer.py

   1 import unittest
   2 from test import test_support
   3
   4 # Silence Py3k warning
   5 test_support.import_module('compiler', deprecated=True)
   6 from compiler import transformer, ast
   7 from compiler import compile
   8
   9 class Tests(unittest.TestCase):
  10
  11     def testMultipleLHS(self):
  12         """ Test multiple targets on the left hand side. """
  13
  14         snippets = ['a, b = 1, 2',
  15                     '(a, b) = 1, 2',
  16                     '((a, b), c) = (1, 2), 3']
  17
  18         for s in snippets:
  19             a = transformer.parse(s)
  20             self.assertIsInstance(a, ast.Module)
  21             child1 = a.getChildNodes()[0]
  22             self.assertIsInstance(child1, ast.Stmt)
  23             child2 = child1.getChildNodes()[0]
  24             self.assertIsInstance(child2, ast.Assign)
  25
  26             # This actually tests the compiler, but it's a way to assure the ast
  27             # is correct
  28             c = compile(s, '<string>', 'single')
  29             vals = {}
  30             exec c in vals
  31             assert vals['a'] == 1
  32             assert vals['b'] == 2
  33
  34 def test_main():
  35     test_support.run_unittest(Tests)
  36
  37 if __name__ == "__main__":
  38     test_main()