from src.codewars.kata_snail import snail


def test_snail():
    array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

    assert snail(array) == [1, 2, 3, 6, 9, 8, 7, 4, 5]

    array = [[1, 2, 3], [8, 9, 4], [7, 6, 5]]
    assert snail(array) == [1, 2, 3, 4, 5, 6, 7, 8, 9]