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

feat: atcoder scraper update for :CP test

Browse source
This commit is contained in:
Barrett Ruth 2025-09-15 18:31:26 -04:00
parent 0fb247f18b
commit c99cf8d4f0
5 changed files with 104 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

26
scrapers/atcoder.py
View file

@ -169,24 +169,26 @@ def main() -> None:
print(json.dumps(result))
sys.exit(1)
test_cases: list[dict[str, str]] = []
for input_data, output_data in tests:
test_cases.append({"input": input_data, "output": output_data})
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
})
if test_cases:
combined_input: str = (
str(len(test_cases))
+ "\n"
+ "\n".join(tc["input"] for tc in test_cases)
)
combined_output: str = "\n".join(tc["output"] for tc in test_cases)
test_cases = [{"input": combined_input, "output": combined_output}]
combined_input = "\n".join(tc["input"] for tc in individual_test_cases)
combined_output = "\n".join(tc["output"] for tc in individual_test_cases)
result = {
"success": True,
"problem_id": problem_id,
"url": url,
"test_cases": test_cases,
"test_cases": individual_test_cases,
"combined": {
"input": combined_input,
"output": combined_output
}
}
print(json.dumps(result))