FilesExpand file tree

 master

/

run_examples.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

93 lines (79 loc) · 2.92 KB

 master

/

run_examples.py

Copy path

Top

File metadata and controls

• Code
• Blame

93 lines (79 loc) · 2.92 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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()