refactor: memoize

This commit is contained in:
Folke Lemaitre 2024-06-11 06:51:30 +02:00
parent 3b447a8037
commit 14d47f650c
No known key found for this signature in database
GPG key ID: 41F8B1FBACAE2040
2 changed files with 42 additions and 7 deletions

36
tests/util/util_spec.lua Normal file
View file

@ -0,0 +1,36 @@
---@module "luassert"
local LazyVim = require("lazyvim.util")
describe("util", function()
local t = 0
local fn = function(a)
t = t + 1
return a
end
local m = LazyVim.memoize(fn)
it("should memoize a function", function()
local a = m(1)
local b = m(1)
local c = m(2)
assert.are.equal(a, b)
assert.are.equal(a, 1)
assert.are.equal(c, 2)
assert.are.equal(t, 2)
assert.are_not.equal(a, c)
end)
local f1 = LazyVim.memoize(function()
return 1
end)
local f2 = LazyVim.memoize(function()
return 2
end)
it("should memoize based on the correct key", function()
assert.are.equal(f1(), 1)
assert.are.equal(f2(), 2)
end)
end)