diff --git a/matrixmania/__init__.py b/matrixmania/__init__.py index c812c86..c71701f 100644 --- a/matrixmania/__init__.py +++ b/matrixmania/__init__.py @@ -1 +1 @@ -from .compute import matmul, transpose, rot_2D, rot_3D \ No newline at end of file +from .compute import matmul, transpose, rot_2D, rot_3D , project_ortho \ No newline at end of file diff --git a/tests/test_projection.py b/tests/test_projection.py index 8715041..dfee0a5 100644 --- a/tests/test_projection.py +++ b/tests/test_projection.py @@ -1 +1,11 @@ from matrixmania import project_ortho + +def test_projection(): + point = (1.0, 2.0, 3.0) + scale = 100 + result1 = project_ortho(point, scale) + result2 = project_ortho(point) + + assert result1 == (100, 200) + assert result2 == (1, 2) +