tangent/lume
1
0
Fork 0
mirror of https://github.com/TangentFoxy/lume.git synced 2024-11-19 07:04:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added lume.pick() and tests

Browse Source
This commit is contained in:
rxi 2015-05-09 15:39:12 +01:00
parent e0d55c8446
commit 044141fefa
2 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lume.lua
View File

@ -422,6 +422,16 @@ function lume.invert(t)
end end
function lume.pick(t, ...)
local rtn = {}
for i = 1, select("#", ...) do
local k = select(i, ...)
rtn[k] = t[k]
end
return rtn
end
function lume.keys(t) function lume.keys(t)
local rtn = {} local rtn = {}
local iter = getiter(t) local iter = getiter(t)

9
test/test_lume.lua
View File

@ -373,6 +373,15 @@ tests["lume.invert"] = function()
testeq( lume.invert(lume.invert{a = 1, b = 2}), {a = 1, b = 2} ) testeq( lume.invert(lume.invert{a = 1, b = 2}), {a = 1, b = 2} )
end end
-- lume.pick
tests["lume.pick"] = function()
local t = { cat = 10, dog = 20, fox = 30, owl = 40 }
testeq( lume.pick(t, "cat", "dog"), { cat = 10, dog = 20 } )
testeq( lume.pick(t, "fox", "owl"), { fox = 30, owl = 40 } )
testeq( lume.pick(t, "owl"), { owl = 40 } )
testeq( lume.pick(t), {} )
end
-- lume.keys -- lume.keys
tests["lume.keys"] = function() tests["lume.keys"] = function()
testeq( lume.keys({}), {} ) testeq( lume.keys({}), {} )