Changeset 1090

Timestamp:

Nov 13, 2009, 11:31:04 PM (14 years ago)

Author:

cmlenz

Message:

Add unit tests for correct handling of with statements and yield expressions in the AST code generator.

Location:

trunk

Files:

• genshi/template/tests/eval.py (modified) (2 diffs)
• scripts/ast_generator.py (modified) (1 diff)

trunk/genshi/template/tests/eval.py

@@ -13 +13 @@
 
 import doctest
+import os
 import pickle
 from StringIO import StringIO
 import sys
+from tempfile import NamedTemporaryFile
 import unittest
 
@@ -784 +786 @@
         self.failIf('k' in d, repr(d))
 
+    if sys.version_info >= (2, 5):
+        def test_with_statement(self):
+            f = NamedTemporaryFile()
+            f.write('foo\nbar\n')
+            f.seek(0)
+
+            d = {'path': f.name}
+            suite = Suite("""from __future__ import with_statement
+lines = []
+with open(path) as file:
+    for line in file:
+        lines.append(line)
+""")
+            suite.execute(d)
+            self.assertEqual(['foo\n', 'bar\n'], d['lines'])
+
+        def test_yield_expression(self):
+            d = {}
+            suite = Suite("""results = []
+def counter(maximum):
+    i = 0
+    while i < maximum:
+        val = (yield i)
+        if val is not None:
+            i = val
+        else:
+            i += 1
+it = counter(5)
+results.append(it.next())
+results.append(it.send(3))
+results.append(it.next())
+""")
+            suite.execute(d)
+            self.assertEqual([0, 3, 4], d['results'])
+
 
 def suite():

trunk/scripts/ast_generator.py

@@ -35 +35 @@
 
 print('# Generated automatically, please do not edit')
-print('# Generator can be found in Genshi SVN, scripts/ast-generator.py')
+print('# Generator can be found in Genshi SVN, scripts/ast_generator.py')
 print('')
 print('__version__ = %s' % _ast.__version__)