diff --git a/tests/gobj/test_lenses.py b/tests/gobj/test_lenses.py new file mode 100644 index 0000000000..a51e497c8e --- /dev/null +++ b/tests/gobj/test_lenses.py @@ -0,0 +1,77 @@ +from panda3d.core import PerspectiveLens, Point3, Point2 + + +def test_perspectivelens_extrude(): + lens = PerspectiveLens() + lens.set_fov(90, 90) + lens.set_near_far(0.5, 100) + + near = Point3() + far = Point3() + + assert lens.extrude((0, 0), near, far) + assert near.almost_equal((0, 0.5, 0), 0.001) + assert far.almost_equal((0, 100, 0), 0.1) + + assert lens.extrude((-1, -1), near, far) + assert near.almost_equal((-0.5, 0.5, -0.5), 0.001) + assert far.almost_equal((-100, 100, -100), 0.1) + + assert lens.extrude((1, 0), near, far) + assert near.almost_equal((0.5, 0.5, 0), 0.001) + assert far.almost_equal((100, 100, 0), 0.1) + + +def test_perspectivelens_extrude_depth(): + lens = PerspectiveLens() + lens.set_fov(90, 90) + lens.set_near_far(0.5, 100) + + point = Point3() + + assert lens.extrude_depth((0, 0, -1), point) + assert point.almost_equal((0, 0.5, 0), 0.001) + + assert lens.extrude_depth((0, 0, 1), point) + assert point.almost_equal((0, 100, 0), 0.001) + + assert lens.extrude_depth((-1, -1, -1), point) + assert point.almost_equal((-0.5, 0.5, -0.5), 0.001) + + assert lens.extrude_depth((-1, -1, 1), point) + assert point.almost_equal((-100, 100, -100), 0.1) + + assert lens.extrude_depth((1, 0, -1), point) + assert point.almost_equal((0.5, 0.5, 0), 0.001) + + assert lens.extrude_depth((1, 0, 1), point) + assert point.almost_equal((100, 100, 0), 0.1) + + +def test_perspectivelens_project(): + lens = PerspectiveLens() + lens.set_fov(90, 90) + lens.set_near_far(0.5, 100) + + point = Point2() + + assert not lens.project((0, 0, 0), point) + assert not lens.project((-1, 0.5, 0), point) + + assert lens.project((0, 0.5, 0), point) + assert point.almost_equal((0, 0), 0.001) + + assert lens.project((0, 100, 0), point) + assert point.almost_equal((0, 0), 0.001) + + assert lens.project((-0.5, 0.5, -0.5), point) + assert point.almost_equal((-1, -1), 0.001) + + assert lens.project((-100, 100, -100), point) + assert point.almost_equal((-1, -1), 0.001) + + assert lens.project((0.5, 0.5, 0), point) + assert point.almost_equal((1, 0), 0.001) + + assert lens.project((100, 100, 0), point) + assert point.almost_equal((1, 0), 0.001) diff --git a/tests/linmath/test_lvector2.py b/tests/linmath/test_lvector2.py new file mode 100644 index 0000000000..f7609ac5d2 --- /dev/null +++ b/tests/linmath/test_lvector2.py @@ -0,0 +1,64 @@ +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 diff --git a/tests/linmath/test_lvector3.py b/tests/linmath/test_lvector3.py new file mode 100644 index 0000000000..016c2e8420 --- /dev/null +++ b/tests/linmath/test_lvector3.py @@ -0,0 +1,74 @@ +from panda3d.core import Vec2, Vec3, Vec3F, Vec3D + + +def test_vec3_creation(): + assert Vec3(x=1, y=2, z=1) == Vec3(1, 2, 1) == Vec3((1, 2, 1)) + + +def test_vec3_getter_setter(): + original_vector = Vec3(2, 3, 7) + + assert original_vector.x == 2 + assert original_vector.y == 3 + assert original_vector.z == 7 + + original_vector.x = 1 + original_vector.y = 3 + original_vector.z = 5 + + assert original_vector == Vec3(1, 3, 5) + + original_vector[0] = 3 + original_vector[1] = 1 + original_vector[2] = 1 + + assert original_vector == Vec3(3, 1, 1) + + original_vector.set_x(-8) + original_vector.set_y(6) + original_vector.set_z(10) + + assert original_vector.x == -8 + assert original_vector.y == 6 + assert original_vector.z == 10 + + +def test_vec3_sum(): + original_vector = Vec3(2, 3, -2) + + assert original_vector + original_vector == Vec3(4, 6, -4) + assert original_vector + 3 == Vec3(5, 6, 1) + + +def test_vec3_power(): + assert Vec3(2, -3, 2) ** 2 == Vec3(4, 9, 4) + + +def test_vec3_len(): + assert len(Vec3(2, -3, 10)) == 3 + + +def test_vec3_swizzle_mask(): + original_vector = Vec3(3, 5, 1) + + assert original_vector.xy == Vec2(3, 5) + assert original_vector.zxy == Vec3(1, 3, 5) + + +def test_vec3_str(): + assert str(Vec3F(2, 3, 1)) == "LVector3f(2, 3, 1)" + assert str(Vec3D(2, 3, 1)) == "LVector3d(2, 3, 1)" + + +def test_vec3_compare(): + assert Vec3(1, 2, 3).compare_to(Vec3(1, 2, 3)) == 0 + + assert Vec3(1, 0, 0).compare_to(Vec3(1, 0, 0)) == 0 + assert Vec3(1, 0, 0).compare_to(Vec3(0, 1, 0)) == 1 + assert Vec3(1, 0, 0).compare_to(Vec3(0, 0, 1)) == 1 + assert Vec3(0, 1, 0).compare_to(Vec3(1, 0, 0)) == -1 + assert Vec3(0, 1, 0).compare_to(Vec3(0, 1, 0)) == 0 + assert Vec3(0, 1, 0).compare_to(Vec3(0, 0, 1)) == 1 + assert Vec3(0, 0, 1).compare_to(Vec3(1, 0, 0)) == -1 + assert Vec3(0, 0, 1).compare_to(Vec3(0, 1, 0)) == -1 + assert Vec3(0, 0, 1).compare_to(Vec3(0, 0, 1)) == 0 diff --git a/tests/linmath/test_lvector4.py b/tests/linmath/test_lvector4.py new file mode 100644 index 0000000000..5a65c7842c --- /dev/null +++ b/tests/linmath/test_lvector4.py @@ -0,0 +1,87 @@ +from panda3d.core import Vec2, Vec3, Vec4, Vec4F, Vec4D + + +def test_vec4_creation(): + assert Vec4(x=1, y=2, z=1, w=7) == Vec4(1, 2, 1, 7) == Vec4((1, 2, 1, 7)) + + +def test_vec4_getter_setter(): + original_vector = Vec4(2, 3, 7, 9) + + assert original_vector.x == 2 + assert original_vector.y == 3 + assert original_vector.z == 7 + assert original_vector.w == 9 + + original_vector.x = 1 + original_vector.y = 3 + original_vector.z = 5 + original_vector.w = -8 + + assert original_vector == Vec4(1, 3, 5, -8) + + original_vector[0] = 3 + original_vector[1] = 1 + original_vector[2] = 1 + original_vector[3] = -2 + + assert original_vector == Vec4(3, 1, 1, -2) + + original_vector.set_x(-8) + original_vector.set_y(6) + original_vector.set_z(10) + original_vector.set_w(30) + + assert original_vector.x == -8 + assert original_vector.y == 6 + assert original_vector.z == 10 + assert original_vector.w == 30 + + +def test_vec4_sum(): + original_vector = Vec4(2, 3, -2, 1) + + assert original_vector + original_vector == Vec4(4, 6, -4, 2) + assert original_vector + 3 == Vec4(5, 6, 1, 4) + + +def test_vec4_power(): + assert Vec4(2, -3, 2, -1) ** 2 == Vec4(4, 9, 4, 1) + + +def test_vec4_len(): + assert len(Vec4(2, -3, 10, 30)) == 4 + + +def test_vec4_swizzle_mask(): + original_vector = Vec4(3, 5, 1, 0) + + assert original_vector.xy == Vec2(3, 5) + assert original_vector.zxy == Vec3(1, 3, 5) + assert original_vector.zxyw == Vec4(1, 3, 5, 0) + + +def test_vec4_str(): + assert str(Vec4F(2, 3, 1, 9)) == "LVector4f(2, 3, 1, 9)" + assert str(Vec4D(2, 3, 1, 9)) == "LVector4d(2, 3, 1, 9)" + + +def test_vec4_compare(): + assert Vec4(1, 2, 3, 4).compare_to(Vec4(1, 2, 3, 4)) == 0 + + assert Vec4(1, 0, 0, 0).compare_to(Vec4(1, 0, 0, 0)) == 0 + assert Vec4(1, 0, 0, 0).compare_to(Vec4(0, 1, 0, 0)) == 1 + assert Vec4(1, 0, 0, 0).compare_to(Vec4(0, 0, 1, 0)) == 1 + assert Vec4(1, 0, 0, 0).compare_to(Vec4(0, 0, 0, 1)) == 1 + assert Vec4(0, 1, 0, 0).compare_to(Vec4(1, 0, 0, 0)) == -1 + assert Vec4(0, 1, 0, 0).compare_to(Vec4(0, 1, 0, 0)) == 0 + assert Vec4(0, 1, 0, 0).compare_to(Vec4(0, 0, 1, 0)) == 1 + assert Vec4(0, 1, 0, 0).compare_to(Vec4(0, 0, 0, 1)) == 1 + assert Vec4(0, 0, 1, 0).compare_to(Vec4(1, 0, 0, 0)) == -1 + assert Vec4(0, 0, 1, 0).compare_to(Vec4(0, 1, 0, 0)) == -1 + assert Vec4(0, 0, 1, 0).compare_to(Vec4(0, 0, 1, 0)) == 0 + assert Vec4(0, 0, 1, 0).compare_to(Vec4(0, 0, 0, 1)) == 1 + assert Vec4(0, 0, 0, 1).compare_to(Vec4(1, 0, 0, 0)) == -1 + assert Vec4(0, 0, 0, 1).compare_to(Vec4(0, 1, 0, 0)) == -1 + assert Vec4(0, 0, 0, 1).compare_to(Vec4(0, 0, 1, 0)) == -1 + assert Vec4(0, 0, 0, 1).compare_to(Vec4(0, 0, 0, 1)) == 0