panda3d/tests/linmath/test_lvector2.py
2020-08-17 12:16:33 +02:00

93 lines
2.3 KiB
Python

from panda3d.core import Vec2, Vec3, Vec4, Vec2F, Vec2D
def test_vec2_creation():
assert Vec2(x=1, y=2) == Vec2(1, 2) == Vec2((1, 2))
def test_vec2_getter_setter():
original_vector = Vec2(2, 3)
assert original_vector.x == 2
assert original_vector.y == 3
original_vector.x = 1
original_vector.y = 3
assert original_vector == Vec2(1, 3)
original_vector[0] = 3
original_vector[1] = 1
assert original_vector == Vec2(3, 1)
original_vector.set_x(-8)
original_vector.set_y(6)
assert original_vector.x == -8
assert original_vector.y == 6
def test_vec2_sum():
original_vector = Vec2(2, 3)
assert original_vector + original_vector == Vec2(4, 6)
assert original_vector + 3 == Vec2(5, 6)
def test_vec2_power():
assert Vec2(2, -3) ** 2 == Vec2(4, 9)
def test_vec2_len():
assert len(Vec2(2, -3)) == 2
def test_vec2_swizzle_mask():
original_vector = Vec2(3, 5)
assert original_vector.yx == Vec2(5, 3)
assert original_vector.xy == original_vector
def test_vec2_str():
assert str(Vec2F(2, 3)) == "LVector2f(2, 3)"
assert str(Vec2D(2, 3)) == "LVector2d(2, 3)"
def test_vec2_compare():
assert Vec2(1, 2).compare_to(Vec2(1, 2)) == 0
assert Vec2(1, 0).compare_to(Vec2(1, 0)) == 0
assert Vec2(1, 0).compare_to(Vec2(0, 1)) == 1
assert Vec2(0, 1).compare_to(Vec2(1, 0)) == -1
assert Vec2(0, 1).compare_to(Vec2(0, 1)) == 0
def test_vec2_nan():
nan = float("nan")
inf = float("inf")
assert not Vec2F(0, 0).is_nan()
assert not Vec2F(1, 0).is_nan()
assert Vec2F(nan, 0).is_nan()
assert Vec2F(0, nan).is_nan()
assert Vec2F(nan, nan).is_nan()
assert Vec2F(-nan, 0).is_nan()
assert Vec2F(-nan, nan).is_nan()
assert Vec2F(inf, nan).is_nan()
assert not Vec2F(inf, 0).is_nan()
assert not Vec2F(inf, inf).is_nan()
assert not Vec2F(-inf, 0).is_nan()
assert not Vec2D(0, 0).is_nan()
assert not Vec2D(1, 0).is_nan()
assert Vec2D(nan, 0).is_nan()
assert Vec2D(0, nan).is_nan()
assert Vec2D(nan, nan).is_nan()
assert Vec2D(-nan, 0).is_nan()
assert Vec2D(-nan, nan).is_nan()
assert Vec2D(inf, nan).is_nan()
assert not Vec2D(inf, 0).is_nan()
assert not Vec2D(inf, inf).is_nan()
assert not Vec2D(-inf, 0).is_nan()