@@ -27,22 +27,12 @@ def clear_directory(dir_path: Path):
2727 file_or_directory .unlink ()
2828
2929
30- async def generate (whitelist : set [ str ], verbose : bool ):
30+ async def generate (verbose : bool ):
3131 test_case_names = set (get_directories (inputs_path )) - {"__pycache__" }
3232
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-
4133 generation_tasks = []
4234 for test_case_name in sorted (test_case_names ):
4335 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
4636 generation_tasks .append (generate_test_case_output (test_case_input_path , test_case_name , verbose ))
4737
4838 failed_test_cases = []
@@ -139,32 +129,13 @@ async def generate_test_case_output(test_case_input_path: Path, test_case_name:
139129 return max (ref_code , plg_code , plg_code_pyd )
140130
141131
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-
156132def main ():
157- if set (sys .argv ).intersection ({"-h" , "--help" }):
158- print (HELP )
159- return
160133 if sys .argv [1 :2 ] == ["-v" ]:
161134 verbose = True
162- whitelist = set (sys .argv [2 :])
163135 else :
164136 verbose = False
165- whitelist = set (sys .argv [1 :])
166137
167- asyncio .run (generate (whitelist , verbose ))
138+ asyncio .run (generate (verbose ))
168139
169140
170141if __name__ == "__main__" :
0 commit comments