diff -r 57875cf423c9 -r 2372284d9457 contrib/benchmarks/revset.py --- a/contrib/benchmarks/revset.py Sat Oct 05 10:29:34 2019 -0400 +++ b/contrib/benchmarks/revset.py Sun Oct 06 09:45:02 2019 -0400 @@ -18,15 +18,16 @@ from . import basedir, perfbench + def createrevsetbenchmark(baseset, variants=None): if variants is None: # Default variants - variants = ["plain", "first", "last", "sort", "sort+first", - "sort+last"] - fname = "track_" + "_".join("".join([ - c if c in string.digits + string.letters else " " - for c in baseset - ]).split()) + variants = ["plain", "first", "last", "sort", "sort+first", "sort+last"] + fname = "track_" + "_".join( + "".join( + [c if c in string.digits + string.letters else " " for c in baseset] + ).split() + ) def wrap(fname, baseset): @perfbench(name=baseset, params=[("variant", variants)]) @@ -36,18 +37,21 @@ for var in variant.split("+"): revset = "%s(%s)" % (var, revset) return perf("perfrevset", revset) + f.__name__ = fname return f + return wrap(fname, baseset) + def initializerevsetbenchmarks(): mod = sys.modules[__name__] - with open(os.path.join(basedir, 'contrib', 'base-revsets.txt'), - 'rb') as fh: + with open(os.path.join(basedir, 'contrib', 'base-revsets.txt'), 'rb') as fh: for line in fh: baseset = line.strip() if baseset and not baseset.startswith('#'): func = createrevsetbenchmark(baseset) setattr(mod, func.__name__, func) + initializerevsetbenchmarks()