mirror of
https://github.com/panda3d/panda3d.git
synced 2026-04-22 22:48:51 -05:00
rdb
126 lines
3.3 KiB
Python
126 lines
3.3 KiB
Python
import pytest
|
|
from copy import copy
|
|
from panda3d import core
|
|
|
|
|
|
def test_mat4_aliases():
|
|
assert core.LMatrix4 is core.Mat4
|
|
assert core.LMatrix4f is core.Mat4F
|
|
assert core.LMatrix4d is core.Mat4D
|
|
|
|
assert (core.LMatrix4f is core.Mat4) != (core.LMatrix4d is core.Mat4)
|
|
|
|
|
|
@pytest.mark.parametrize("type", (core.LMatrix4d, core.LMatrix4f))
|
|
def test_mat4_constructor(type):
|
|
# Test that three ways of construction produce the same matrix.
|
|
mat1 = type((1, 2, 3, 4),
|
|
(5, 6, 7, 8),
|
|
(9, 10, 11, 12),
|
|
(13, 14, 15, 16))
|
|
|
|
mat2 = type(1, 2, 3, 4,
|
|
5, 6, 7, 8,
|
|
9, 10, 11, 12,
|
|
13, 14, 15, 16)
|
|
|
|
mat3 = type((1, 2, 3, 4,
|
|
5, 6, 7, 8,
|
|
9, 10, 11, 12,
|
|
13, 14, 15, 16))
|
|
|
|
assert mat1 == mat2
|
|
assert mat2 == mat3
|
|
assert mat1 == mat3
|
|
|
|
|
|
@pytest.mark.parametrize("type", (core.LMatrix4d, core.LMatrix4f))
|
|
def test_mat4_copy_constuctor(type):
|
|
mat1 = type((1, 2, 3, 4),
|
|
(5, 6, 7, 8),
|
|
(9, 10, 11, 12),
|
|
(13, 14, 15, 16))
|
|
|
|
# Make a copy. Changing it should not change the original.
|
|
mat2 = type(mat1)
|
|
assert mat1 == mat2
|
|
mat2[0][0] = 100
|
|
assert mat1 != mat2
|
|
|
|
# Make a copy by unpacking.
|
|
mat2 = type(*mat1)
|
|
assert mat1 == mat2
|
|
mat2[0][0] = 100
|
|
assert mat1 != mat2
|
|
|
|
# Make a copy by calling copy.copy.
|
|
mat2 = copy(mat1)
|
|
assert mat1 == mat2
|
|
mat2[0][0] = 100
|
|
assert mat1 != mat2
|
|
|
|
|
|
@pytest.mark.parametrize("type", (core.LMatrix4d, core.LMatrix4f))
|
|
def test_mat4_invert_same_type(type):
|
|
mat = type((1, 0, 0, 0,
|
|
0, 1, 0, 0,
|
|
0, 0, 1, 0,
|
|
1, 2, 3, 1))
|
|
|
|
inv = core.invert(mat)
|
|
assert mat.__class__ == inv.__class__
|
|
|
|
|
|
@pytest.mark.parametrize("type", (core.LMatrix4d, core.LMatrix4f))
|
|
def test_mat4_invert_correct(type):
|
|
mat = type((1, 0, 0, 0,
|
|
0, 1, 0, 0,
|
|
0, 0, 1, 0,
|
|
1, 2, 3, 1))
|
|
inv = type()
|
|
assert inv.invert_from(mat)
|
|
|
|
assert inv == type(( 1, 0, 0, 0,
|
|
0, 1, 0, 0,
|
|
0, 0, 1, 0,
|
|
-1, -2, -3, 1))
|
|
|
|
assert (mat * inv).is_identity()
|
|
assert (inv * mat).is_identity()
|
|
|
|
|
|
@pytest.mark.parametrize("type", (core.LMatrix4d, core.LMatrix4f))
|
|
def test_mat4_rows(type):
|
|
mat = type((1, 2, 3, 4,
|
|
5, 6, 7, 8,
|
|
9, 10, 11, 12,
|
|
13, 14, 15, 16))
|
|
|
|
assert mat.rows[0] == (1, 2, 3, 4)
|
|
assert mat.rows[1] == (5, 6, 7, 8)
|
|
assert mat.rows[2] == (9, 10, 11, 12)
|
|
assert mat.rows[3] == (13, 14, 15, 16)
|
|
|
|
assert mat.get_row3(0) == (1, 2, 3)
|
|
assert mat.get_row3(1) == (5, 6, 7)
|
|
assert mat.get_row3(2) == (9, 10, 11)
|
|
assert mat.get_row3(3) == (13, 14, 15)
|
|
|
|
|
|
@pytest.mark.parametrize("type", (core.LMatrix4d, core.LMatrix4f))
|
|
def test_mat4_cols(type):
|
|
mat = type((1, 5, 9, 13,
|
|
2, 6, 10, 14,
|
|
3, 7, 11, 15,
|
|
4, 8, 12, 16))
|
|
|
|
assert mat.cols[0] == (1, 2, 3, 4)
|
|
assert mat.cols[1] == (5, 6, 7, 8)
|
|
assert mat.cols[2] == (9, 10, 11, 12)
|
|
assert mat.cols[3] == (13, 14, 15, 16)
|
|
|
|
assert mat.get_col3(0) == (1, 2, 3)
|
|
assert mat.get_col3(1) == (5, 6, 7)
|
|
assert mat.get_col3(2) == (9, 10, 11)
|
|
assert mat.get_col3(3) == (13, 14, 15)
|