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
|
#!/usr/bin/env python3
"""self-hosted run-only images retain enhancements and generic method bodies."""
import sys
import argparse
import subprocess
import tempfile
from pathlib import Path
sys.dont_write_bytecode = True # importing imagekit must not leave __pycache__ in the tree
from imagekit import micro_image
parser = argparse.ArgumentParser()
parser.add_argument('--case', default='')
parser.add_argument('--host', default='build/bin/zy')
parser.add_argument('--mkgenome', default='build/bin/mkgenome')
parser.add_argument('--map', default='build/core/map/memmap.h')
args = parser.parse_args()
cases = [
('enhancements', 'include rt; fn main() { rt.put_n(17 % 5); rt.put_c(32); rt.put_n(17 / 5); }', b'2 3'),
('array algorithms', Path('examples/learn/31-array-algorithms.zc').read_text(), Path('examples/learn/31-array-algorithms.out').read_bytes()),
('owned pop', 'include rt; int*.u take() { int*[?] a = rt.new(int*[?]); int* p = rt.new(int); *p = 8; a.push(p); return a.pop(); } fn main() { let p = take(); rt.put_n(rt.isobj(p)); rt.put_c(32); rt.put_n(*p); }', b'1 8'),
]
prefix = b''.join(Path(f).read_bytes() + b'\n' for f in ['src/runtime/rt.zc', 'src/core/core.zc', 'src/runtime/run.zc'])
for name, source, expected in cases:
if args.case and args.case not in name:
continue
with tempfile.TemporaryDirectory(prefix='tinyvm-image-') as work:
directory = Path(work)
r = subprocess.run([args.host], input=Path('src/tools/build.zc').read_bytes() + b'`' + prefix + source.encode(), capture_output=True, timeout=30)
assert r.returncode == 0 and len(r.stdout) >= 56 and not r.stdout.startswith(b'zc:'), (name, r.stdout[:1000], r.stderr)
image = subprocess.run([args.mkgenome], input=r.stdout, capture_output=True, timeout=5)
assert image.returncode == 0, (name, image.stderr)
binary = micro_image(directory, image.stdout, args.map)
r = subprocess.run([str(binary)], capture_output=True, timeout=10)
assert r.returncode == 0 and r.stdout == expected, (name, r.returncode, r.stdout, r.stderr, expected)
print('ok self-hosted run-only image: ' + name)
|
