Version 2

matrixmania/__init__.py

@@ -1 +1 @@
-from .compute import matmul, transpose, rot_2D
+from .compute import matmul, transpose, rot_2D, rot_3D

matrixmania/compute.py

@@ -127,3 +127,9 @@ if __name__ == "__main__":
     print("Matrix A:")
     print(tabulate(matrix_a))
 
+    theta = math.pi / 2
+    R = rot_3D(theta, "y")
+
+    print("rotation matrix for 90° on the y axis:")
+    print(tabulate(R))
+