yo

test/test_matrixmania/test_rot.py

@@ -0,0 +1,88 @@
+import math
+import pytest
+from src.MatrixMania.compute import rot_2d, rot_3d
+
+
+# --- Tests für rot_2d ---
+
+def test_rot_2d_vector_valid():
+    # 90° Drehung von [1, 0] um den Ursprung -> [0, 1]
+    res = rot_2d([1.0, 0.0], math.pi / 2)
+    assert math.isclose(res[0], 0.0, abs_tol=1e-9)
+    assert math.isclose(res[1], 1.0, abs_tol=1e-9)
+
+
+def test_rot_2d_matrix_valid():
+    # 90° Drehung im Uhrzeigersinn
+    m = [[1, 2], [3, 4]]
+    assert rot_2d(m) == [[3, 1], [4, 2]]
+
+
+def test_rot_2d_invalid_vector():
+    # Vektor mit ungültiger Länge muss ValueError werfen
+    with pytest.raises(ValueError):
+        rot_2d([1.0, 2.0, 3.0], math.pi)
+
+
+# --- Tests für rot_3D ---
+
+def test_rot_3d_identity():
+    # Bei Winkel 0 müssen alle Achsen die Einheitsmatrix liefern
+    eye = [
+        [1.0, 0.0, 0.0],
+        [0.0, 1.0, 0.0],
+        [0.0, 0.0, 1.0],
+    ]
+    for axis in ["x", "y", "z"]:
+        res = rot_3d(0.0, axis)
+        for r in range(3):
+            for c in range(3):
+                assert math.isclose(res[r][c], eye[r][c], abs_tol=1e-9)
+
+
+def test_rot_3d_x_axis():
+    # 90° um x-Achse
+    res = rot_3d(math.pi / 2, "x")
+    expected = [
+        [1.0, 0.0, 0.0],
+        [0.0, 0.0, -1.0],
+        [0.0, 1.0, 0.0],
+    ]
+    for r in range(3):
+        for c in range(3):
+            assert math.isclose(res[r][c], expected[r][c], abs_tol=1e-9)
+
+
+def test_rot_3d_y_axis():
+    # 90° um y-Achse
+    res = rot_3d(math.pi / 2, "y")
+    expected = [
+        [0.0, 0.0, 1.0],
+        [0.0, 1.0, 0.0],
+        [-1.0, 0.0, 0.0],
+    ]
+    for r in range(3):
+        for c in range(3):
+            assert math.isclose(res[r][c], expected[r][c], abs_tol=1e-9)
+
+
+def test_rot_3d_z_axis():
+    # 90° um z-Achse
+    res = rot_3d(math.pi / 2, "z")
+    expected = [
+        [0.0, -1.0, 0.0],
+        [1.0, 0.0, 0.0],
+        [0.0, 0.0, 1.0],
+    ]
+    for r in range(3):
+        for c in range(3):
+            assert math.isclose(res[r][c], expected[r][c], abs_tol=1e-9)
+
+
+def test_rot_3d_invalid_axis():
+    # Ungültige Achsenbezeichnung muss ValueError werfen
+    with pytest.raises(ValueError):
+        rot_3d(math.pi / 2, "w")
+
+    with pytest.raises(ValueError):
+        rot_3d(math.pi / 2, "1")