fix formatting;

This commit is contained in:
Barrett Ruth 2025-09-12 17:20:38 -05:00
parent 0db6fa96b4
commit 03807e46e0
4 changed files with 15 additions and 14 deletions

View file

@ -21,7 +21,6 @@ local function get_plugin_path()
return vim.fn.fnamemodify(plugin_path, ":h:h:h")
end
local competition_types = { "atcoder", "codeforces", "cses" }
local function setup_contest(contest_type)
@ -58,7 +57,7 @@ local function setup_problem(problem_id, problem_letter)
vim.g.cp_diff_mode = false
end
vim.cmd('silent only')
vim.cmd("silent only")
local scrape_result = scrape.scrape_problem(vim.g.cp_contest, problem_id, problem_letter)

View file

@ -90,20 +90,20 @@ function M.scrape_problem(contest, problem_id, problem_letter)
if #data.test_cases > 0 then
local all_inputs = {}
local all_outputs = {}
for i, test_case in ipairs(data.test_cases) do
local input_lines = vim.split(test_case.input:gsub("\r", ""):gsub("\n+$", ""), "\n")
local output_lines = vim.split(test_case.output:gsub("\r", ""):gsub("\n+$", ""), "\n")
for _, line in ipairs(input_lines) do
table.insert(all_inputs, line)
end
for _, line in ipairs(output_lines) do
table.insert(all_outputs, line)
end
end
vim.fn.writefile(all_inputs, input_file)
vim.fn.writefile(all_outputs, expected_file)
end

View file

@ -35,7 +35,7 @@ function M.restore_layout(state)
end
vim.cmd.diffoff()
local problem_id = vim.fn.expand("%:t:r")
if problem_id == "" then
for win, win_state in pairs(state.windows) do
@ -50,13 +50,13 @@ function M.restore_layout(state)
end
if problem_id ~= "" then
vim.cmd('silent only')
vim.cmd("silent only")
local base_fp = vim.fn.getcwd()
local input = ("%s/io/%s.in"):format(base_fp, problem_id)
local output = ("%s/io/%s.out"):format(base_fp, problem_id)
local source = problem_id .. ".cc"
vim.cmd.edit(source)
vim.cmd.vsplit(output)
vim.bo.filetype = "cp"
@ -68,7 +68,7 @@ function M.restore_layout(state)
vim.cmd.wincmd("h")
else
vim.cmd(state.layout)
for win, win_state in pairs(state.windows) do
if vim.api.nvim_win_is_valid(win) then
vim.api.nvim_set_current_win(win)
@ -86,7 +86,7 @@ end
function M.setup_diff_layout(actual_output, expected_output, input_file)
vim.cmd.diffoff()
vim.cmd('silent only')
vim.cmd("silent only")
local output_lines = vim.split(actual_output, "\n")
local output_buf = vim.api.nvim_create_buf(false, true)

View file

@ -88,9 +88,11 @@ def main():
test_cases = []
for input_data, output_data in tests:
test_cases.append({"input": input_data, "output": output_data})
if test_cases:
combined_input = str(len(test_cases)) + "\n" + "\n".join(tc["input"] for tc in test_cases)
combined_input = (
str(len(test_cases)) + "\n" + "\n".join(tc["input"] for tc in test_cases)
)
combined_output = "\n".join(tc["output"] for tc in test_cases)
test_cases = [{"input": combined_input, "output": combined_output}]