Run 2to3 on doctest source.

1 files changed, 5 insertions, 4 deletions

diff --git a/setup.py b/setup.py
index 3a47a09c..d91ea069 100755
--- a/setup.py
+++ b/setup.py
@@ -18,11 +18,12 @@ if sys.version_info >= (3,):
         outf, copied = file_util.copy_file(f, os.path.join(tmp_src, f), update=1)
         if copied and outf.endswith(".py"):
             outfiles_2to3.append(outf)
-    util.run_2to3(outfiles_2to3)
+        if copied and outf.endswith('api_tests.txt'):
+            # XXX support this in distutils as well
+            from lib2to3.main import main
+            main('lib2to3.fixes', ['-wd', os.path.join(tmp_src, 'tests', 'api_tests.txt')])
 
-    # XXX support this in distutils as well
-    from lib2to3.main import main
-    main('lib2to3.fixes', ['-wd', os.path.join(tmp_src, 'tests', 'api_tests.txt')])
+    util.run_2to3(outfiles_2to3)
 
     # arrange setup to use the copy
     sys.path.insert(0, tmp_src)