FilesExpand file tree

 parse-bytecode

/

tinytest-codegen.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

65 lines (52 loc) · 1.98 KB

 parse-bytecode

/

tinytest-codegen.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

65 lines (52 loc) · 1.98 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

#! /usr/bin/env python3

import os, sys

from glob import glob

from re import sub

def escape(s):

lookup = {

'\0': '\\0',

'\t': '\\t',

'\n': '\\n\"\n\"',

'\r': '\\r',

'\\': '\\\\',

'\"': '\\\"',

}

return "\"\"\n\"{}\"".format(''.join([lookup[x] if x in lookup else x for x in s]))

def chew_filename(t):

return { 'func': "test_{}_fn".format(sub(r'/|\.|-', '_', t)), 'desc': t.split('/')[1] }

def script_to_map(t):

r = { 'name': chew_filename(t)['func'] }

with open(t) as f: r['script'] = escape(''.join(f.readlines()))

return r

test_function = (

"void {name}(void* data) {{\n"

" const char * pystr = {script};\n"

" do_str(pystr);\n"

"}}"

)

testcase_struct = (

"struct testcase_t {name}_tests[] = {{\n{body}\n END_OF_TESTCASES\n}};"

)

testcase_member = (

" {{ \"{desc}\", {func}, TT_ENABLED_, 0, 0 }},"

)

testgroup_struct = (

"struct testgroup_t groups[] = {{\n{body}\n END_OF_GROUPS\n}};"

)

testgroup_member = (

" {{ \"{name}/\", {name}_tests }},"

)

## XXX: may be we could have `--without <groups>` argument...

# currently these tests are selected because they pass on qemu-arm

test_dirs = ('basics', 'micropython', 'inlineasm') # 'float', 'import', 'io', 'misc')

exclude_tests = ('inlineasm/asmfpaddsub.py', 'inlineasm/asmfpcmp.py', 'inlineasm/asmfpldrstr.py', 'inlineasm/asmfpmuldiv.py', 'inlineasm/asmfpsqrt.py',)

output = []

for group in test_dirs:

tests = [test for test in glob('{}/*.py'.format(group)) if test not in exclude_tests]

output.extend([test_function.format(**script_to_map(test)) for test in tests])

testcase_members = [testcase_member.format(**chew_filename(test)) for test in tests]

output.append(testcase_struct.format(name=group, body='\n'.join(testcase_members)))

testgroup_members = [testgroup_member.format(name=group) for group in test_dirs]

output.append(testgroup_struct.format(body='\n'.join(testgroup_members)))

## XXX: may be we could have `--output <filename>` argument...

print('\n\n'.join(output))