Skip to content
This repository was archived by the owner on Jun 9, 2025. It is now read-only.

Commit 2944111

Browse files
Simplify generation (#61)
* Remove help code * Remove whitelist
1 parent b5a4314 commit 2944111

File tree

1 file changed

+2
-31
lines changed

1 file changed

+2
-31
lines changed

tests/generate.py

Lines changed: 2 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -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-
156132
def 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

170141
if __name__ == "__main__":

0 commit comments

Comments
 (0)