summaryrefslogtreecommitdiffstats
path: root/Lib/lib2to3/tests/benchmark.py
blob: e1b293fd89080486ae819db8590e960b35654c81 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#!/usr/bin/env python2.5
"""
This is a benchmarking script to test the speed of 2to3's pattern matching
system. It's equivalent to "refactor.py -f all" for every Python module
in sys.modules, but without engaging the actual transformations.
"""

__author__ = "Collin Winter <collinw at gmail.com>"

# Python imports
import os.path
import sys
from time import time

# Test imports
from .support import adjust_path
adjust_path()

# Local imports
from .. import refactor

### Mock code for refactor.py and the fixers
###############################################################################
class Options:
    def __init__(self, **kwargs):
        for k, v in list(kwargs.items()):
            setattr(self, k, v)

        self.verbose = False

def dummy_transform(*args, **kwargs):
    pass

### Collect list of modules to match against
###############################################################################
files = []
for mod in list(sys.modules.values()):
    if mod is None or not hasattr(mod, '__file__'):
        continue
    f = mod.__file__
    if f.endswith('.pyc'):
        f = f[:-1]
    if f.endswith('.py'):
        files.append(f)

### Set up refactor and run the benchmark
###############################################################################
options = Options(fix=["all"], print_function=False, doctests_only=False)
refactor = refactor.RefactoringTool(options)
for fixer in refactor.fixers:
    # We don't want them to actually fix the tree, just match against it.
    fixer.transform = dummy_transform

t = time()
for f in files:
    print("Matching", f)
    refactor.refactor_file(f)
print("%d seconds to match %d files" % (time() - t, len(sys.modules)))