leef-math-cd2025/spec/vec2_spec.lua
2016-08-21 20:42:44 -07:00

186 lines
4.2 KiB
Lua

local vec2 = require "modules.vec2"
local DBL_EPSILON = require("modules.constants").DBL_EPSILON
local abs, sqrt = math.abs, math.sqrt
describe("vec2:", function()
it("creates an empty vector", function()
local a = vec2()
assert.is.equal(0, a.x)
assert.is.equal(0, a.y)
assert.is_true(a:is_vec2())
assert.is_true(a:is_zero())
end)
it("creates a vector from a number", function()
local a = vec2(3)
assert.is.equal(3, a.x)
assert.is.equal(3, a.y)
end)
it("creates a vector from numbers", function()
local a = vec2(3, 5)
assert.is.equal(3, a.x)
assert.is.equal(5, a.y)
end)
it("creates a vector from a list", function()
local a = vec2 { 3, 5 }
assert.is.equal(3, a.x)
assert.is.equal(5, a.y)
end)
it("creates a vector from a record", function()
local a = vec2 { x=3, y=5 }
assert.is.equal(3, a.x)
assert.is.equal(5, a.y)
end)
it("clones a vector", function()
local a = vec2(3, 5)
local b = a:clone()
assert.is.equal(a, b)
end)
it("adds a vector to another", function()
local a = vec2(3, 5)
local b = vec2(7, 4)
local c = vec2():add(a, b)
local d = a + b
assert.is.equal(10, c.x)
assert.is.equal(9, c.y)
assert.is.equal(c, d)
end)
it("subracts a vector from another", function()
local a = vec2(3, 5)
local b = vec2(7, 4)
local c = vec2():sub(a, b)
local d = a - b
assert.is.equal(-4, c.x)
assert.is.equal( 1, c.y)
assert.is.equal( c, d)
end)
it("multiplies a vector by a scale factor", function()
local a = vec2(3, 5)
local s = 2
local c = vec2():scale(a, s)
local d = a * s
assert.is.equal(6, c.x)
assert.is.equal(10, c.y)
assert.is.equal(c, d)
end)
it("divides a vector by another vector", function()
local a = vec2(3, 5)
local s = vec2(2, 2)
local c = vec2():div(a, s)
local d = a / s
assert.is.equal(1.5, c.x)
assert.is.equal(2.5, c.y)
assert.is.equal(c, d)
end)
it("inverts a vector", function()
local a = vec2(3, -5)
local b = -a
assert.is.equal(-a.x, b.x)
assert.is.equal(-a.y, b.y)
end)
it("gets the length of a vector", function()
local a = vec2(3, 5)
assert.is.equal(sqrt(34), a:len())
end)
it("gets the square length of a vector", function()
local a = vec2(3, 5)
assert.is.equal(34, a:len2())
end)
it("normalizes a vector", function()
local a = vec2(3, 5)
local b = vec2():normalize(a)
assert.is_true(abs(b:len()-1) < DBL_EPSILON)
end)
it("trims the length of a vector", function()
local a = vec2(3, 5)
local b = vec2():trim(a, 0.5)
assert.is_true(abs(b:len()-0.5) < DBL_EPSILON)
end)
it("gets the distance between two vectors", function()
local a = vec2(3, 5)
local b = vec2(7, 4)
local c = a:dist(b)
assert.is.equal(sqrt(17), c)
end)
it("gets the square distance between two vectors", function()
local a = vec2(3, 5)
local b = vec2(7, 4)
local c = a:dist2(b)
assert.is.equal(17, c)
end)
it("crosses two vectors", function()
local a = vec2(3, 5)
local b = vec2(7, 4)
local c = a:cross(b)
assert.is.equal(-23, c)
end)
it("dots two vectors", function()
local a = vec2(3, 5)
local b = vec2(7, 4)
local c = a:dot(b)
assert.is.equal(41, c)
end)
it("interpolates between two vectors", function()
local a = vec2(3, 5)
local b = vec2(7, 4)
local s = 0.1
local c = vec2():lerp(a, b, s)
assert.is.equal(3.4, c.x)
assert.is.equal(4.9, c.y)
end)
it("unpacks a vector", function()
local a = vec2(3, 5)
local x, y = a:unpack()
assert.is.equal(3, x)
assert.is.equal(5, y)
end)
it("rotates a vector", function()
local a = vec2(3, 5)
local b = vec2():rotate(a, math.pi)
local c = vec2():rotate(b, -math.pi)
assert.is_not.equal(a, b)
assert.is.equal(a, c)
end)
it("converts between polar and cartesian coordinates", function()
local a = vec2(3, 5)
local r, t = a:to_polar()
local b = vec2.from_cartesian(r, t)
assert.is.equal(a.x, b.x)
assert.is.equal(a.y, b.y)
end)
it("gets a perpendicular vector", function()
local a = vec2(3, 5)
local b = vec2():perpendicular(a)
assert.is.equal(-5, b.x)
assert.is.equal( 3, b.y)
end)
it("gets a string representation of a vector", function()
local a = vec2()
local b = a:to_string()
assert.is.equal("(+0.000,+0.000)", b)
end)
end)