Module:Exponential search/testcases

local search = require('Module:Exponential search')
local ScribuntoUnit = require('Module:ScribuntoUnit')
local suite = ScribuntoUnit:new()

local function makeArrayTest(len)
	local array = {}
	for i = 1, len do
		array[i] = true
	end
	return function (i)
		return array[i]
	end
end

local arrayTests = {}
setmetatable(arrayTests, {
	__index = function (t, key)
		local func = makeArrayTest(key)
		arrayTests[key] = func
		return func
	end
})

function suite:assertError(func, ...)
	local success, ret = pcall(func, ...)
	self:assertFalse(success)
	return ret
end

function suite:assertErrorMsg(msg, func, ...)
	local ret = self:assertError(func, ...)
	self:assertStringContains(msg, ret, true)
end

function suite:testFuncType()
	self:assertError(search, 5)
	self:assertError(search, 'foo')
	self:assertError(search, true)
	self:assertError(search, {})
	self:assertError(search, nil)
end

function suite:testInitType()
	self:assertError(search, arrayTests[0], 'foo')
	self:assertError(search, arrayTests[0], arrayTests[0])
	self:assertError(search, arrayTests[0], true)
	self:assertError(search, arrayTests[0], {})
	self:assertError(search, arrayTests[1], 'foo')
end
 
function suite:testInitIntegerCheck()
	self:assertErrorMsg(
		"invalid init value '0' detected in argument #2 to 'Exponential search' (init values must be a positive integer)",
		search, arrayTests[0], 0
	)
	self:assertErrorMsg(
		"invalid init value '1.5' detected in argument #2 to 'Exponential search' (init values must be a positive integer)",
		search, arrayTests[0], 1.5
	)
	self:assertErrorMsg(
		"invalid init value '-3' detected in argument #2 to 'Exponential search' (init values must be a positive integer)",
		search, arrayTests[0], -3
	)
	self:assertErrorMsg(
		"invalid init value 'inf' detected in argument #2 to 'Exponential search' (init values must be a positive integer)",
		search, arrayTests[0], math.huge
	)
	self:assertErrorMsg(
		"invalid init value '-nan' detected in argument #2 to 'Exponential search' (init values must be a positive integer)",
		search, arrayTests[0], 0/0
	)
end

function suite:testKeyNotFound()
	self:assertEquals(nil, search(arrayTests[0]))
end

function suite:testKeyNotFoundInit()
	self:assertEquals(nil, search(arrayTests[0], 5))
end

function suite:test1()
	self:assertEquals(1, search(arrayTests[1]))
end

function suite:test1Init1()
	self:assertEquals(1, search(arrayTests[1], 1))
end

function suite:test1Init2()
	self:assertEquals(1, search(arrayTests[1], 2))
end

function suite:test1Init3()
	self:assertEquals(1, search(arrayTests[1], 3))
end

function suite:test2()
	self:assertEquals(2, search(arrayTests[2]))
end

function suite:test2Init1()
	self:assertEquals(2, search(arrayTests[2], 1))
end

function suite:test2Init2()
	self:assertEquals(2, search(arrayTests[2], 2))
end

function suite:test2Init3()
	self:assertEquals(2, search(arrayTests[2], 3))
end

function suite:test3()
	self:assertEquals(3, search(arrayTests[3]))
end

function suite:test3Init1()
	self:assertEquals(3, search(arrayTests[3], 1))
end

function suite:test3Init2()
	self:assertEquals(3, search(arrayTests[3], 2))
end

function suite:test3Init3()
	self:assertEquals(3, search(arrayTests[3], 3))
end

function suite:testLargeArray()
	self:assertEquals(12345, search(arrayTests[12345]))
end

function suite:testLargeInit()
	self:assertEquals(3, search(arrayTests[3], 12345))
end

return suite