fix formatting;
This commit is contained in:
parent
0db6fa96b4
commit
03807e46e0
4 changed files with 15 additions and 14 deletions
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
|
|
@ -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}]
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue