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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
import os
import subprocess
import shutil
import tempfile
from pathlib import Path, PurePath
from term_color import Color, colored
import threading
from my_typing import *
PassFail = NamedTuple('PassFail',
[('passed', bool),
('reason', str),
('tag', Optional[str]),
('stderr', Optional[str]),
('stdout', Optional[str]),
('hc_opts', Optional[str]),
])
def badResult(result: PassFail) -> bool:
return not result.passed
def passed(hc_opts=None) -> PassFail:
return PassFail(passed=True,
reason='',
tag=None,
stderr=None,
stdout=None,
hc_opts=hc_opts)
def failBecause(reason: str,
tag: str=None,
stderr: str=None,
stdout: str=None
) -> PassFail:
return PassFail(passed=False, reason=reason, tag=tag,
stderr=stderr, stdout=stdout, hc_opts=None)
def strip_quotes(s: str) -> str:
# Don't wrap commands to subprocess.call/Popen in quotes.
return s.strip('\'"')
def str_warn(s: str) -> str:
return colored(Color.YELLOW, s)
def str_info(s: str) -> str:
return colored(Color.BLUE, s)
def getStdout(cmd_and_args: List[str]):
# Can't use subprocess.check_output, since we also verify that
# no stderr was produced
p = subprocess.Popen([strip_quotes(cmd_and_args[0])] + cmd_and_args[1:],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
(stdout, stderr) = p.communicate()
r = p.wait()
if r != 0:
raise Exception("Command failed: " + str(cmd_and_args))
if stderr:
raise Exception("stderr from command: %s\nOutput:\n%s\n" % (cmd_and_args, stderr.decode('utf-8')))
return stdout.decode('utf-8')
def lndir(srcdir: Path, dstdir: Path):
# Create symlinks for all files in src directory.
# Not all developers might have lndir installed.
# os.system('lndir -silent {0} {1}'.format(srcdir, dstdir))
for filename in srcdir.iterdir():
base = filename.relative_to(srcdir)
src = srcdir / base
dst = dstdir / base
if src.is_file():
link_or_copy_file(src, dst)
else:
dst.mkdir()
lndir(src, dst)
# All possible test metric strings.
def testing_metrics():
return ['bytes allocated', 'peak_megabytes_allocated', 'max_bytes_used']
# On Windows, os.symlink is not defined with Python 2.7, but is in Python 3
# when using msys2, as GHC does. Unfortunately, only Administrative users have
# the privileges necessary to create symbolic links by default. Consequently we
# are forced to just copy instead.
#
# We define the following function to make this magic more
# explicit/discoverable. You are encouraged to use it instead of os.symlink.
def symlinks_work() -> bool:
if os.getenv('FORCE_SYMLINKS') is not None:
return True
elif os.name == 'nt':
# On Windows we try to create a symlink to test whether symlinks are
# usable.
works = False
with tempfile.NamedTemporaryFile() as tmp:
try:
tmp.write(b'hello')
os.symlink(tmp.name, '__symlink-test')
works = True
except OSError as e:
print('Saw {} during symlink test; assuming symlinks do not work.'.format(e))
finally:
try:
os.unlink('__symlink-test')
except:
pass
return works
else:
return True
if not symlinks_work():
def link_or_copy_file(src: Path, dst: Path):
shutil.copyfile(str(src), str(dst))
else:
def link_or_copy_file(src: Path, dst: Path):
os.symlink(str(src), str(dst))
class Watcher(object):
def __init__(self, count: int) -> None:
self.pool = count
self.evt = threading.Event()
self.sync_lock = threading.Lock()
if count <= 0:
self.evt.set()
def wait(self):
self.evt.wait()
def notify(self):
self.sync_lock.acquire()
self.pool -= 1
if self.pool <= 0:
self.evt.set()
self.sync_lock.release()
def memoize(f):
"""
A decorator to memoize a nullary function.
"""
def cached():
if cached._cache is None:
cached._cache = f()
return cached._cache
cached._cache = None
return cached
# Print the matrix data in a tabular format.
def print_table(header_rows: List[List[str]], data_rows: List[List[str]], padding=2) -> None:
# Calculate column widths then print each row.
colWidths = [(0 if idx == 0 else padding) + max([len(cell) for cell in col])
for (idx, col) in enumerate(zip(*(header_rows + data_rows)))]
col_fmts = ['{:>' + str(w) + '}' for w in colWidths]
def printCols(cols):
for row in cols:
print(''.join([f.format(cell) for (f,cell) in zip(col_fmts, row)]))
printCols(header_rows)
print('-' * sum(colWidths))
printCols(data_rows)
def shorten_metric_name(name: str) -> str:
dic = {
"runtime/bytes allocated": "run/alloc",
"runtime/peak_megabytes_allocated": "run/peak",
"runtime/max_bytes_used": "run/max",
"compile_time/bytes allocated": "ghc/alloc",
"compile_time/peak_megabytes_allocated": "ghc/peak",
"compile_time/max_bytes_used": "ghc/max",
}
return dic.get(name, name)
|
