from matrixmania import rot_2D
import math

def test_rot_1():
    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]]