tests: add unit tests for lens extrude and project operations

2025-09-29 16:20:11 -04:00 · 2019-03-29 12:20:01 +01:00 · 2019-03-29 12:20:01 +01:00 · 5a8a8dd550
commit 5a8a8dd550
parent b06b5791ca
1 changed files with 77 additions and 0 deletions
--- a/tests/gobj/test_lenses.py
+++ 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)