[Python-checkins] python/dist/src/Lib/test test_compiler.py, 1.12, 1.13

[nascheme@users.sourceforge.net]

Update of /cvsroot/python/python/dist/src/Lib/test
In directory sc8-pr-cvs1.sourceforge.net:/tmp/cvs-serv367/Lib/test

Modified Files:
	test_compiler.py 
Log Message:
Fix compiler.ast.flatten function so that it works on lists.


Index: test_compiler.py
===================================================================
RCS file: /cvsroot/python/python/dist/src/Lib/test/test_compiler.py,v
retrieving revision 1.12
retrieving revision 1.13
diff -u -d -r1.12 -r1.13
--- test_compiler.py	20 Apr 2005 17:45:12 -0000	1.12
+++ test_compiler.py	2 Jun 2005 05:55:19 -0000	1.13
@@ -1,4 +1,5 @@
 import compiler
+from compiler.ast import flatten
 import os
 import test.test_support
 import unittest
@@ -60,6 +61,10 @@
         for child in node.getChildNodes():
             self.check_lineno(child)
 
+    def testFlatten(self):
+        self.assertEquals(flatten([1, [2]]), [1, 2])
+        self.assertEquals(flatten((1, (2,))), [1, 2])
+
 NOLINENO = (compiler.ast.Module, compiler.ast.Stmt, compiler.ast.Discard)
 
 ###############################################################################