AppPkg/Applications/Python/Python-2.7.2/Lib/lib2to3/tests/support.py

   1 """Support code for test_*.py files"""
   2 # Author: Collin Winter
   3
   4 # Python imports
   5 import unittest
   6 import sys
   7 import os
   8 import os.path
   9 import re
  10 from textwrap import dedent
  11
  12 # Local imports
  13 from lib2to3 import pytree, refactor
  14 from lib2to3.pgen2 import driver
  15
  16 test_dir = os.path.dirname(__file__)
  17 proj_dir = os.path.normpath(os.path.join(test_dir, ".."))
  18 grammar_path = os.path.join(test_dir, "..", "Grammar.txt")
  19 grammar = driver.load_grammar(grammar_path)
  20 driver = driver.Driver(grammar, convert=pytree.convert)
  21
  22 def parse_string(string):
  23     return driver.parse_string(reformat(string), debug=True)
  24
  25 def run_all_tests(test_mod=None, tests=None):
  26     if tests is None:
  27         tests = unittest.TestLoader().loadTestsFromModule(test_mod)
  28     unittest.TextTestRunner(verbosity=2).run(tests)
  29
  30 def reformat(string):
  31     return dedent(string) + u"\n\n"
  32
  33 def get_refactorer(fixer_pkg="lib2to3", fixers=None, options=None):
  34     """
  35     A convenience function for creating a RefactoringTool for tests.
  36
  37     fixers is a list of fixers for the RefactoringTool to use. By default
  38     "lib2to3.fixes.*" is used. options is an optional dictionary of options to
  39     be passed to the RefactoringTool.
  40     """
  41     if fixers is not None:
  42         fixers = [fixer_pkg + ".fixes.fix_" + fix for fix in fixers]
  43     else:
  44         fixers = refactor.get_fixers_from_package(fixer_pkg + ".fixes")
  45     options = options or {}
  46     return refactor.RefactoringTool(fixers, options, explicit=True)
  47
  48 def all_project_files():
  49     for dirpath, dirnames, filenames in os.walk(proj_dir):
  50         for filename in filenames:
  51             if filename.endswith(".py"):
  52                 yield os.path.join(dirpath, filename)
  53
  54 TestCase = unittest.TestCase