barrett/cp.nvim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 13

feat(scraper): update scraper i/o for single test panel

Browse source
This commit is contained in:
Barrett Ruth 2025-09-18 09:56:54 -04:00
parent 24624f9826
commit b2e1ea2c58
3 changed files with 108 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

15
scrapers/atcoder.py
View file

@ -169,21 +169,16 @@ def main() -> None:
print(json.dumps(result))
sys.exit(1)
individual_test_cases: list[dict[str, str]] = []
for index, (input_data, output_data) in enumerate(tests, 1):
individual_test_cases.append(
{"index": index, "input": input_data, "output": output_data}
)
combined_input = "\n".join(tc["input"] for tc in individual_test_cases)
combined_output = "\n".join(tc["output"] for tc in individual_test_cases)
test_list: list[dict[str, str]] = []
for input_data, output_data in tests:
normalized_input = "1\n" + input_data
test_list.append({"input": normalized_input, "expected": output_data})
result = {
"success": True,
"problem_id": problem_id,
"url": url,
"test_cases": individual_test_cases,
"combined": {"input": combined_input, "output": combined_output},
"tests": test_list,
}
print(json.dumps(result))