Add initial test suite.

See #3 for details.

Closes #4.

.travis.yml

+language: python
+python:
+  - "pypy"
+  - "2.6"
+  - "2.7"
+  - "3.2"
+  - "3.3"
+  - "3.4"
+cache:
+  - $HOME/.pip-cache
+install:
+  - pip install -r requirements-dev.txt --download-cache $HOME/.pip-cache
+  - if [[ $TRAVIS_PYTHON_VERSION == '2.6' ]]; then pip install unittest2; fi
+  - pip install --allow-all-external -e .
+script: nosetests

requirements-dev.txt

-wheel==0.23.0
+wheel>=0.23.0
+nose>=1.3.0
+tox>=1.7.1
+flake8>=2.1.0

tests/nested_with_for_tests.py

-class A(object):
-    pass
-
-class B(object):
-    pass
-
-def test():
-    with A() as a:
-        with B() as b:
-            print ("test1")
-
-def test2():
-    with A() as a, B() as b:
-        print ("test2")

tests/test_codegen.py

-#!/usr/bin/env python
 """
 Part of the astor library for Python AST manipulation
 
 License: 3-clause BSD
 
-Copyright 2012 (c) Patrick Maupin
+Copyright 2014 (c) Berker Peksag
 """
 
-
-import fnmatch
-import os
-import sys
 import ast
+import unittest
+import textwrap
 
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 import astor
 
 
-def findpy(root):
-    if isinstance(root, str):
-        if os.path.isfile(root):
-            if root.endswith('.py'):
-                yield root
-            return
-        for fname in os.listdir(root):
-            fname = os.path.join(root, fname)
-            if os.path.isdir(fname):
-                for fname in findpy(fname):
-                    yield fname
-            elif fnmatch.fnmatch(fname, '*.py'):
-                yield fname
-    else:
-        for dirname in root:
-            for fname in findpy(dirname):
-                yield fname
+class CodegenTestCase(unittest.TestCase):
+
+    def assertAstSourceEqual(self, source):
+        self.assertEqual(astor.to_source(ast.parse(source)), source)
+
+    def test_imports(self):
+        source = "import ast"
+        self.assertAstSourceEqual(source)
+        source = "import operator as op"
+        self.assertAstSourceEqual(source)
+        source = "from math import floor"
+        self.assertAstSourceEqual(source)
 
-def testone(fname, f1=None, f2=None):
-    try:
-        ast1 = astor.parsefile(fname)
-    except (SyntaxError, UnicodeDecodeError):
-        print("IGNORED %s" % fname)
-        return
-    dump1 = astor.dump(ast1)
-    reconstitute = '# -*- coding: utf-8 -*-\n' + astor.to_source(ast1)
-    ast2 = ast.parse(reconstitute, fname)
-    dump2 = astor.dump(ast2)
-    ok = dump1 == dump2
-    print('%-8s%s' % ('OK' if dump1 == dump2 else 'FAIL', fname))
-    if not ok and f1 is not None and f2 is not None:
-        f1.write('\n\n***************************************\n%s\n***************************************\n\n\n' % fname)
-        f2.write('\n\n***************************************\n%s\n***************************************\n\n\n' % fname)
-        f1.write(dump1)
-        f2.write(dump2)
-        f = open('bad.txt', 'w')
-        f.write(reconstitute)
-        f.close()
-    return ok
+    def test_try_expect(self):
+        source = textwrap.dedent("""\
+        try:
+            'spam'[10]
+        except IndexError:
+            pass""")
+        self.assertAstSourceEqual(source)
 
+        source = textwrap.dedent("""\
+        try:
+            'spam'[10]
+        except IndexError as exc:
+            sys.stdout.write(exc)""")
+        self.assertAstSourceEqual(source)
 
-def go(root, stoponfail=True, f1=None, f2=None):
-    for fname in findpy(root or os.path.dirname(fnmatch.__file__)):
-        if not testone(fname, f1, f2) and stoponfail:
-            break
+    def test_del_statement(self):
+        source = "del l[0]"
+        self.assertAstSourceEqual(source)
+        source = "del obj.x"
+        self.assertAstSourceEqual(source)
 
 
 if __name__ == '__main__':
-    f1 = open('test_codegen_dump_1.txt', 'w')
-    f2 = open('test_codegen_dump_2.txt', 'w')
-    go(sys.argv[1:], False, f1, f2)
+    unittest.main()

tests/test_transform.py

-#!/usr/bin/env python
-
-import sys
-import os
-sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-import astor
-
-def TestMe(x, y, width=10, foo=None):
-    from foo.bar.baz.murgatroyd import sally as bob
-
-    a.b = c.d + x.y.z.a.b
-    m.n = q = (w.w, x.x.y.y) = f(x.x.y.z)
-
-func_ast = astor.codetoast(TestMe)
-print(astor.dump(func_ast))
-print(astor.to_source(func_ast))
-
-class ConsolidateAttributes(astor.TreeWalk):
-    def post_Attribute(self):
-        node = self.cur_node
-        value = node.value
-        value.id += '.' + node.attr
-        self.replace(value)
-
-ConsolidateAttributes(func_ast)
-
-class FindNames(astor.TreeWalk):
-    def init_Assign(self):
-        self.assignments = []
-        self.current_assign = None
-    def pre_Assign(self):
-        self.current_assign = [], []
-    def post_Assign(self):
-        self.assignments.append(self.current_assign)
-        self.current_assign = None
-
-    def pre_targets_name(self):
-        self.in_targets = True
-    def post_targets_name(self):
-        self.in_targets = False
-    def post_Name(self):
-        if self.current_assign:
-            self.current_assign[self.in_targets].append(self.cur_node.id)
-
-x = FindNames(func_ast)
-print(x.assignments)
-
-
-print('')
-print('')
-print(astor.dump(func_ast))
-print(astor.to_source(func_ast))

tox.ini

+[tox]
+envlist = py26,py27,pypy,py32,py33,py34
+skipsdist = True
+
+[testenv]
+commands =
+ pip install --allow-all-external -e .
+ nosetests
+ flake8 astor tests
+deps =
+ -rrequirements-dev.txt
+
+[testenv:py26]
+deps =
+ {[testenv]deps}
+ unittest2