FilesExpand file tree

 6.2.x

/

tinytest-codegen.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

126 lines (103 loc) · 4.22 KB

 6.2.x

/

tinytest-codegen.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

126 lines (103 loc) · 4.22 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

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

#!/usr/bin/env python3

# SPDX-FileCopyrightText: 2014 MicroPython & CircuitPython contributors (https://github.com/adafruit/circuitpython/graphs/contributors)

#

# SPDX-License-Identifier: MIT

import os, sys

from glob import glob

from re import sub

import argparse

def escape(s):

s = s.decode()

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}

def script_to_map(test_file):

r = {"name": chew_filename(test_file)["func"]}

with open(test_file, "rb") as test:

script = test.readlines()

# Test for import skip_if and inject it into the test as needed.

if "import skip_if\n" in script:

index = script.index("import skip_if\n")

script.pop(index)

script.insert(index, "class skip_if:\n")

with open("../tests/skip_if.py") as skip_if:

total_lines = 1

for line in skip_if:

stripped = line.strip()

if not stripped or stripped.startswith(("#", '"""')):

continue

script.insert(index + total_lines, "\t" + line)

total_lines += 1

r["script"] = escape(b"".join(script))

with open(test_file + ".exp", "rb") as f:

r["output"] = escape(f.read())

return r

test_function = (

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

" static const char pystr[] = {script};\n"

" static const char exp[] = {output};\n"

" upytest_set_expected_output(exp, sizeof(exp) - 1);\n"

" upytest_execute_test(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", "float", "extmod", "inlineasm") # 'import', 'io', 'misc')

exclude_tests = (

# pattern matching in .exp

"basics/bytes_compare3.py",

"extmod/ticks_diff.py",

"extmod/time_ms_us.py",

"extmod/uheapq_timeq.py",

# unicode char issue

"extmod/ujson_loads.py",

# doesn't output to python stdout

"extmod/ure_debug.py",

"extmod/vfs_basic.py",

"extmod/vfs_fat_ramdisk.py",

"extmod/vfs_fat_fileio.py",

"extmod/vfs_fat_fsusermount.py",

"extmod/vfs_fat_oldproto.py",

# rounding issues

"float/float_divmod.py",

# requires double precision floating point to work

"float/float2int_doubleprec_intbig.py",

"float/float_parse_doubleprec.py",

# inline asm FP tests (require Cortex-M4)

"inlineasm/asmfpaddsub.py",

"inlineasm/asmfpcmp.py",

"inlineasm/asmfpldrstr.py",

"inlineasm/asmfpmuldiv.py",

"inlineasm/asmfpsqrt.py",

# different filename in output

"micropython/emg_exc.py",

"micropython/heapalloc_traceback.py",

# pattern matching in .exp

"micropython/meminfo.py",

)

output = []

tests = []

argparser = argparse.ArgumentParser(

description="Convert native MicroPython tests to tinytest/upytesthelper C code"

)

argparser.add_argument("--stdin", action="store_true", help="read list of tests from stdin")

args = argparser.parse_args()

if not args.stdin:

for group in test_dirs:

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

else:

for l in sys.stdin:

tests.append(l.rstrip())

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="", body="\n".join(testcase_members)))

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

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

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

# Don't depend on what system locale is set, use utf8 encoding.

sys.stdout.buffer.write("\n\n".join(output).encode("utf8"))