mirrors/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2025-10-21 17:21:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(iter): add Iter.take (#26525)

Browse Source
This commit is contained in:
Will Hopkins
2023-12-12 12:27:24 -08:00
committed by GitHub
parent 1907abb4c2
commit 69ffbb76c2
3 changed files with 81 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
test/functional/lua/iter_spec.lua
View File

@@ -203,6 +203,33 @@ describe('vim.iter', function()
matches('skipback%(%) requires a list%-like table', pcall_err(it.nthback, it, 1))
end)
it('take()', function()
do
local t = { 4, 3, 2, 1 }
eq({}, vim.iter(t):take(0):totable())
eq({ 4 }, vim.iter(t):take(1):totable())
eq({ 4, 3 }, vim.iter(t):take(2):totable())
eq({ 4, 3, 2 }, vim.iter(t):take(3):totable())
eq({ 4, 3, 2, 1 }, vim.iter(t):take(4):totable())
eq({ 4, 3, 2, 1 }, vim.iter(t):take(5):totable())
end
do
local t = { 4, 3, 2, 1 }
local it = vim.iter(t)
eq({ 4, 3 }, it:take(2):totable())
-- tail is already set from the previous take()
eq({ 4, 3 }, it:take(3):totable())
end
do
local it = vim.iter(vim.gsplit('a|b|c|d', '|'))
eq({ 'a', 'b' }, it:take(2):totable())
-- non-array iterators are consumed by take()
eq({}, it:take(2):totable())
end
end)
it('any()', function()
local function odd(v)
return v % 2 ~= 0