From f5afa78ff92180ae90b83e35e1cf662e6afde5e9 Mon Sep 17 00:00:00 2001 From: Mona Benkert Date: Tue, 4 Nov 2025 17:14:36 +0100 Subject: [PATCH] new tests --- test/test_rot.py | 9 ++++++++- test/test_transpose.py | 8 ++++++++ 2 files changed, 16 insertions(+), 1 deletion(-) create mode 100644 test/test_transpose.py diff --git a/test/test_rot.py b/test/test_rot.py index 4190ed3..6626872 100644 --- a/test/test_rot.py +++ b/test/test_rot.py @@ -1,4 +1,11 @@ from matrixmania import rot_2D +import math def test_rot_1(): - angle = \ No newline at end of file + angle = 0 + assert rot_2D(angle) == [[1.0, -0.0], + [0.0, 1.0]] + +def test_rot_2(): + angle = math.pi + assert rot_2D(angle) == [[-1.0, -1.2246467991473532e-16], [1.2246467991473532e-16, -1.0]] \ No newline at end of file diff --git a/test/test_transpose.py b/test/test_transpose.py new file mode 100644 index 0000000..64c00b2 --- /dev/null +++ b/test/test_transpose.py @@ -0,0 +1,8 @@ +from matrixmania import transpose + +def test_transpose_1(): + a = [[1, 2, 3], + [4, 5, 6]] + assert transpose(a) == [[1, 4], + [2, 5], + [3, 6]] \ No newline at end of file