diff --git a/matrixmania/__init__.py b/matrixmania/__init__.py index 975a585..c812c86 100644 --- a/matrixmania/__init__.py +++ b/matrixmania/__init__.py @@ -1 +1 @@ -from .compute import matmul, transpose, rot_2D \ No newline at end of file +from .compute import matmul, transpose, rot_2D, rot_3D \ No newline at end of file diff --git a/tests/test_rot3D.py b/tests/test_rot3D.py index ad408f3..65d073e 100644 --- a/tests/test_rot3D.py +++ b/tests/test_rot3D.py @@ -1 +1,9 @@ -from matrixmania import rot_3D \ No newline at end of file +from matrixmania import rot_3D + +def test_rot_3D(): + angle = 90 + result = rot_3D(angle,"x") + print(f"Angle:{angle} Rot-Matrix:{result}") + + assert result == [[1, 0, 0], [0, 6.123233995736766e-17, -1.0], [0, 1.0, 6.123233995736766e-17]] == [[6.123233995736766e-17, -1.0], [1.0, 6.123233995736766e-17]] +