from matrixmania import transpose

def test_transpose():
    matrix = [
        [1, 2, 3],
        [4, 5, 6]
    ]
    matrix = transpose(matrix)

    assert matrix ==  [[1, 4],[2, 5],[3, 6]]