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

feat(run_panel): max_output_lines

Browse source
This commit is contained in:
Barrett Ruth 2025-09-19 19:40:50 -04:00
parent c7338d01d8
commit f22eccfa89
2 changed files with 23 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
lua/cp/test.lua
View file

@ -118,7 +118,7 @@ end
---@param contest_config ContestConfig
---@param test_case TestCase
---@return table
local function run_single_test_case(ctx, contest_config, test_case)
local function run_single_test_case(ctx, contest_config, cp_config, test_case)
local language = vim.fn.fnamemodify(ctx.source_file, ':e')
local language_name = constants.filetype_to_language[language] or contest_config.default_language
local language_config = contest_config[language_name]
@ -187,6 +187,18 @@ local function run_single_test_case(ctx, contest_config, test_case)
local execution_time = (vim.uv.hrtime() - start_time) / 1000000
local actual_output = (result.stdout or ''):gsub('\n$', '')
local max_lines = cp_config.run_panel.max_output_lines
local output_lines = vim.split(actual_output, '\n')
if #output_lines > max_lines then
local trimmed_lines = {}
for i = 1, max_lines do
table.insert(trimmed_lines, output_lines[i])
end
table.insert(trimmed_lines, string.format('... (output trimmed after %d lines)', max_lines))
actual_output = table.concat(trimmed_lines, '\n')
end
local expected_output = test_case.expected:gsub('\n$', '')
local ok = actual_output == expected_output
@ -238,7 +250,7 @@ end
---@param contest_config ContestConfig
---@param index number
---@return boolean
function M.run_test_case(ctx, contest_config, index)
function M.run_test_case(ctx, contest_config, cp_config, index)
local test_case = run_panel_state.test_cases[index]
if not test_case then
return false