Module:array/testcases

From Wiktionary, the free dictionary
Jump to navigation Jump to search

24 out of 24 assertions passed


local export = {}

local Array = require "Module:array"

local function collect(iter, state, k)
	local vals = Array()
	for val in iter, state, k do
		vals:insert(val)
	end
	return vals
end

local total_tests, successes = 0, 0
local function assert(val)
	total_tests = total_tests + 1
	if val then
		successes = successes + 1
	end
end

function export.run_tests()
	assert(Array(1, 2, 3):map(function (x) return x * x end):concat ", " == "1, 4, 9")
	
	assert(Array(1, 2, 3, 4, 5):filter(function (x) return x % 2 == 0 end):concat ", " == "2, 4")
	
	assert(Array():type() == "array")
	
	assert(Array(1, nil, 2, nil, 3):compress():concat ", " == "1, 2, 3")
	assert(collect(Array(1, nil, 2, nil, 3):sparseIpairs()):concat ", " == "1, 3, 5")
	
	assert(Array("a", "b", "c"):toSet()["a"] == true)
	assert(Array("a", "b", "c"):to_set()["a"] == true)
	
	assert(Array(2, 4, 8):all(function (x) return x % 2 == 0 end) == true)
	assert(Array(1, 2, 3):some(function (x) return x % 2 == 0 end) == true)
	
	assert(Array(1, 2, 3, nil, 4):length() == 3)
	
	assert(Array(1, 2, 3, nil, 4):contains(3) == true)
	assert(Array(1, 2, 3, nil, 4):contains(4) == false)
	
	assert(Array.keysToList(Array("a", "b", "c"):invert()):concat ", " == "a, b, c")
	
	assert(Array(1, nil, 2, nil, 3):maxIndex() == 5)
	assert(Array(1, nil, 2, nil, 3):max_index() == 5)
	
	assert(Array(1, nil, 2, nil, 3):num_keys():concat ", " == "1, 3, 5")
	
	assert(collect(Array("a", "b", "c"):reverseIpairs()):concat ", " == "3, 2, 1")
	
	assert(Array("a", "b", "c"):reverse():concat ", " == "c, b, a")
	
	assert(Array("a", "b", "c"):serialCommaJoin{ dontTag = true } == "a, b, and c")
	
	assert(Array(1, 1, 2, 3, 1, 2, 3):removeDuplicates():concat ", " == "1, 2, 3")
	
	assert(Array(1, 2, 3, 4, 5):slice(1, 3):concat ", " == "1, 2, 3")
	assert(Array(1, 2, 3, 4, 5):slice(1, -3):concat ", " == "1, 2, 3")
	
	assert(Array(1, 2, 3):adjust_index(-1) == 3)
	
	assert((Array(1, 2, 3) + Array(4, 5, 6)):concat ", " == "1, 2, 3, 4, 5, 6")
	
	local color = successes < total_tests and "red" or "green"
	
	return ('<span style="color: %s;">%d out of %d assertions passed</span>')
		:format(color, successes, total_tests)
end

return export