FilesExpand file tree

 main

/

tinytest-codegen.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

executable file

·

157 lines (134 loc) · 4.91 KB

 main

/

tinytest-codegen.py

Copy path

Top

File metadata and controls

• Code
• Blame

executable file

·

157 lines (134 loc) · 4.91 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

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

#!/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"

' printf("\\n");\n'

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

" upytest_execute_test(pystr);\n"

' printf("result: ");\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",

"misc",

"extmod",

"float",

"inlineasm",

"qemu-arm",

) # 'import', 'io',)

exclude_tests = (

# pattern matching in .exp

"basics/bytes_compare3.py",

"extmod/ticks_diff.py",

"extmod/time_ms_us.py",

# unicode char issue

"extmod/json_loads.py",

# doesn't output to python stdout

"extmod/re_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_format_ints_doubleprec.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",

# don't have emergency exception buffer

"micropython/heapalloc_exc_compressed_emg_exc.py",

# pattern matching in .exp

"micropython/meminfo.py",

# needs sys stdfiles

"misc/print_exception.py",

# settrace .exp files are too large

"misc/sys_settrace_loop.py",

"misc/sys_settrace_generator.py",

"misc/sys_settrace_features.py",

# don't have f-string

"basics/string_fstring.py",

"basics/string_fstring_debug.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")

argparser.add_argument("--exclude", action="append", help="exclude test by name")

args = argparser.parse_args()

if not args.stdin:

if args.exclude:

exclude_tests += tuple(args.exclude)

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