Mercurial > genshi > genshi-test

diff genshi/template/tests/eval.py @ 808:a134e21a8f44 stable-0.5.x
Ported [1008] and [1009] to 0.5.x branch.
author: cmlenz
date: Fri, 06 Mar 2009 11:53:07 +0000
parents: cce33406c1cf
--- a/genshi/template/tests/eval.py
+++ b/genshi/template/tests/eval.py
@@ -572,6 +572,16 @@
         suite.execute(_ctxt2dict(data))
         assert 'ifilter' in data
 
+    def test_import_in_def(self):
+        suite = Suite("""def fun():
+    from itertools import ifilter
+    return ifilter(None, xrange(3))
+""")
+        data = Context()
+        suite.execute(data)
+        assert 'ifilter' not in data
+        self.assertEqual([1, 2], list(data['fun']()))
+
     def test_for(self):
         suite = Suite("""x = []
 for i in range(3):