Skip to content
This repository was archived by the owner on Feb 2, 2024. It is now read-only.

Commit 83681fc

Browse files
authored
Create test case without exec (#595)
1 parent d529471 commit 83681fc

File tree

1 file changed

+6
-15
lines changed

1 file changed

+6
-15
lines changed

sdc/tests/tests_perf/generator.py

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -94,27 +94,18 @@ def gen_call_expr(test_case, prefix):
9494

9595

9696
def gen_test(test_case, prefix):
97-
func_name = 'func'
98-
9997
usecase = gen_usecase(test_case, prefix)
10098

101-
skip = '@skip_numba_jit\n' if test_case.skip else ''
102-
10399
test_name = test_case.name
104100
if test_case.params:
105-
test_name = f'{test_name}({test_case.params})'
101+
test_name += f'({test_case.params})'
106102

107-
func_text = f"""
108-
{skip}def {func_name}(self):
109-
self._test_case(usecase, name='{test_name}', total_data_length={test_case.size},
110-
data_num={test_case.data_num}, input_data={test_case.input_data})
111-
"""
103+
def func(self):
104+
self._test_case(usecase, name=test_name, total_data_length=test_case.size,
105+
data_num=test_case.data_num, input_data=test_case.input_data)
112106

113-
loc_vars = {}
114-
global_vars = {'usecase': usecase,
115-
'skip_numba_jit': skip_numba_jit}
116-
exec(func_text, global_vars, loc_vars)
117-
func = loc_vars[func_name]
107+
if test_case.skip:
108+
func = skip_numba_jit(func)
118109

119110
return func
120111

0 commit comments

Comments
 (0)