mirrors/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2026-04-29 10:44:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf(treesitter): more efficient foldexpr

Browse Source
This commit is contained in:
Lewis Russell
2023-03-09 15:28:55 +00:00
parent c5b9643bf1
commit 46b73bf22c
3 changed files with 259 additions and 124 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
test/functional/treesitter/parser_spec.lua
View File

@@ -888,18 +888,20 @@ int x = INT_MAX;
it("can fold via foldexpr", function()
insert(test_text)
local levels = exec_lua([[
vim.opt.filetype = 'c'
vim.treesitter.get_parser(0, "c")
local res = {}
for i = 1, vim.api.nvim_buf_line_count(0) do
res[i] = vim.treesitter.foldexpr(i)
end
return res
]])
local function get_fold_levels()
return exec_lua([[
local res = {}
for i = 1, vim.api.nvim_buf_line_count(0) do
res[i] = vim.treesitter.foldexpr(i)
end
return res
]])
end
exec_lua([[vim.treesitter.get_parser(0, "c")]])
eq({
[1] = '>1',
[1] = '>1',
[2] = '1',
[3] = '1',
[4] = '1',
@@ -917,6 +919,33 @@ int x = INT_MAX;
[16] = '3',
[17] = '3',
[18] = '2',
[19] = '1' }, levels)
[19] = '1' }, get_fold_levels())
helpers.command('1,2d')
helpers.poke_eventloop()
exec_lua([[vim.treesitter.get_parser():parse()]])
helpers.poke_eventloop()
helpers.sleep(100)
eq({
[1] = '0',
[2] = '0',
[3] = '>1',
[4] = '1',
[5] = '1',
[6] = '0',
[7] = '0',
[8] = '>1',
[9] = '1',
[10] = '1',
[11] = '1',
[12] = '1',
[13] = '>2',
[14] = '2',
[15] = '2',
[16] = '1',
[17] = '0' }, get_fold_levels())
end)
end)