Fix compiler.ast.flatten function so that it works on lists.

This commit is contained in:
Neil Schemenauer 2005-06-02 05:55:20 +00:00
parent 76276177de
commit f36947032f
3 changed files with 13 additions and 8 deletions

View file

@ -1,4 +1,5 @@
import compiler
from compiler.ast import flatten
import os
import test.test_support
import unittest
@ -60,6 +61,10 @@ class CompilerTest(unittest.TestCase):
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)
###############################################################################