@@ -27,22 +27,12 @@ def clear_directory(dir_path: Path):
27
27
file_or_directory .unlink ()
28
28
29
29
30
- async def generate (whitelist : set [ str ], verbose : bool ):
30
+ async def generate (verbose : bool ):
31
31
test_case_names = set (get_directories (inputs_path )) - {"__pycache__" }
32
32
33
- path_whitelist = set ()
34
- name_whitelist = set ()
35
- for item in whitelist :
36
- if item in test_case_names :
37
- name_whitelist .add (item )
38
- continue
39
- path_whitelist .add (item )
40
-
41
33
generation_tasks = []
42
34
for test_case_name in sorted (test_case_names ):
43
35
test_case_input_path = inputs_path .joinpath (test_case_name ).resolve ()
44
- if whitelist and str (test_case_input_path ) not in path_whitelist and test_case_name not in name_whitelist :
45
- continue
46
36
generation_tasks .append (generate_test_case_output (test_case_input_path , test_case_name , verbose ))
47
37
48
38
failed_test_cases = []
@@ -139,32 +129,13 @@ async def generate_test_case_output(test_case_input_path: Path, test_case_name:
139
129
return max (ref_code , plg_code , plg_code_pyd )
140
130
141
131
142
- HELP = "\n " .join (
143
- (
144
- "Usage: python generate.py [-h] [-v] [DIRECTORIES or NAMES]" ,
145
- "Generate python classes for standard tests." ,
146
- "" ,
147
- "DIRECTORIES One or more relative or absolute directories of test-cases to generateclasses for." ,
148
- " python generate.py inputs/bool inputs/double inputs/enum" ,
149
- "" ,
150
- "NAMES One or more test-case names to generate classes for." ,
151
- " python generate.py bool double enums" ,
152
- ),
153
- )
154
-
155
-
156
132
def main ():
157
- if set (sys .argv ).intersection ({"-h" , "--help" }):
158
- print (HELP )
159
- return
160
133
if sys .argv [1 :2 ] == ["-v" ]:
161
134
verbose = True
162
- whitelist = set (sys .argv [2 :])
163
135
else :
164
136
verbose = False
165
- whitelist = set (sys .argv [1 :])
166
137
167
- asyncio .run (generate (whitelist , verbose ))
138
+ asyncio .run (generate (verbose ))
168
139
169
140
170
141
if __name__ == "__main__" :
0 commit comments