mond läuft

compute.py

@@ -1,17 +1,7 @@
-"""Matrix multiplication with explicit dimension checks.
-
-This module provides a pure‑Python implementation that
-- validates input dimensions and structure
-- performs generic matrix multiplication without external libs
-- raises informative ``ValueError`` when inputs are inconsistent.
-
-The implementation is small but fully typed so it can be
-imported into other projects or used as a teaching example.
-"""
-
 from __future__ import annotations
 
 from collections.abc import Sequence
+import math
 
 Matrix = Sequence[Sequence[float]]
 
@@ -67,6 +57,23 @@ def matmul(left: Matrix, right: Matrix) -> list[list[float]]:
 
 
     # Helper to validate a matrix is non‑empty and rectangular.
+def rot_2D(theta: float) -> list[list[float]]:
+    """Return the 2‑D rotation matrix for a given angle.
+
+    Parameters
+    ----------
+    theta
+        Rotation angle in radians.
+
+    Returns
+    -------
+    list[list[float]]
+        2×2 rotation matrix ``[[cosθ, -sinθ], [sinθ, cosθ]]``.
+    """
+    c = math.cos(theta)
+    s = math.sin(theta)
+    return [[c, -s], [s, c]]
+
 def _validate(matrix: Matrix, name: str) -> None:
     """Check that *matrix* is a non‑empty, rectangular 2D sequence.
 
@@ -158,55 +165,10 @@ if __name__ == "__main__":
         matmul([[1, 2, 3]], [[1, 2]])
     except ValueError as error:
         print("example 3 passed:", error)
-else:
-    raise AssertionError("example 3 failed: no ValueError raised")
 
-if __name__ == "__main__":
-    # Example 1: standard 2x3 * 3x2 multiplication.
-    a = [
-        [1, 2, 3],
-        [4, 5, 6],
-    ]
-    b = [
-        [7, 8],
-        [9, 10],
-        [11, 12],
-    ]
-    expected = [
-        [58, 64],
-        [139, 154],
-    ]
-    assert matmul(a, b) == expected, f"example 1 failed: {matmul(a, b)}"
-    print("example 1 passed:", matmul(a, b))
 
-    # Example 2: multiplication with the identity matrix changes nothing.
-    identity = [
-        [1, 0, 0],
-        [0, 1, 0],
-        [0, 0, 1],
-    ]
-    c = [
-        [2, -1, 0],
-        [0, 3, 4],
-        [5, 6, -2],
-    ]
-    assert matmul(c, identity) == c, "example 2 failed"
-    print("example 2 passed:", matmul(c, identity))
+    # end of main guard
 
-    # Example 3: mismatched dimensions must raise ValueError.
-    try:
-        matmul([[1, 2, 3]], [[1, 2]])
-    except ValueError as error:
-        print("example 3 passed:", error)
-    else:
-        raise AssertionError("example 3 failed: no ValueError raised")
-
-    # Example 4: Demonstration of the transpose function.
-    example_matrix = [[1, 2, 3], [4, 5, 6]]
-    transposed = transpose(example_matrix)
-    print("Example 4: transpose([[1,2,3],[4,5,6]]) =", transposed)
-    assert transposed == [[1, 4], [2, 5], [3, 6]], "Example 4 failed"
-    print("example 4 passed")