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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
"""
Run and time some or all examples.
"""
from argparse import ArgumentParser
from contextlib import ExitStack
import os
from pathlib import Path
import subprocess
import sys
from tempfile import TemporaryDirectory
import time
import tokenize
_preamble = """\
from matplotlib import pyplot as plt
def pseudo_show(block=True):
for num in plt.get_fignums():
plt.figure(num).savefig(f"{num}")
plt.show = pseudo_show
"""
class RunInfo:
def __init__(self, backend, elapsed, failed):
self.backend = backend
self.elapsed = elapsed
self.failed = failed
def __str__(self):
s = ""
if self.backend:
s += f"{self.backend}: "
s += f"{self.elapsed}ms"
if self.failed:
s += " (failed!)"
return s
def main():
parser = ArgumentParser(description=__doc__)
parser.add_argument(
"--backend", action="append",
help=("backend to test; can be passed multiple times; defaults to the "
"default backend"))
parser.add_argument(
"--include-sgskip", action="store_true",
help="do not filter out *_sgskip.py examples")
parser.add_argument(
"--rundir", type=Path,
help=("directory from where the tests are run; defaults to a "
"temporary directory"))
parser.add_argument(
"paths", nargs="*", type=Path,
help="examples to run; defaults to all examples (except *_sgskip.py)")
args = parser.parse_args()
root = Path(__file__).resolve().parent.parent / "examples"
paths = args.paths if args.paths else sorted(root.glob("**/*.py"))
if not args.include_sgskip:
paths = [path for path in paths if not path.stem.endswith("sgskip")]
relpaths = [path.resolve().relative_to(root) for path in paths]
width = max(len(str(relpath)) for relpath in relpaths)
for relpath in relpaths:
print(str(relpath).ljust(width + 1), end="", flush=True)
runinfos = []
with ExitStack() as stack:
if args.rundir:
cwd = args.rundir / relpath.with_suffix("")
cwd.mkdir(parents=True)
else:
cwd = stack.enter_context(TemporaryDirectory())
with tokenize.open(root / relpath) as src:
Path(cwd, relpath.name).write_text(
_preamble + src.read(), encoding="utf-8")
for backend in args.backend or [None]:
env = {**os.environ}
if backend is not None:
env["MPLBACKEND"] = backend
start = time.perf_counter()
proc = subprocess.run([sys.executable, relpath.name],
cwd=cwd, env=env)
elapsed = round(1000 * (time.perf_counter() - start))
runinfos.append(RunInfo(backend, elapsed, proc.returncode))
print("\t".join(map(str, runinfos)))
if __name__ == "__main__":
main()
|
