diff --git a/src/codewars/kata_snail.py b/src/codewars/kata_snail.py new file mode 100644 index 0000000..5e7d5fe --- /dev/null +++ b/src/codewars/kata_snail.py @@ -0,0 +1,6 @@ +def snail(snail_map): + return snail_map + + +array = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] +print(snail(array)) diff --git a/tests/test_codewars/test_snail.py b/tests/test_codewars/test_snail.py new file mode 100644 index 0000000..61cf0db --- /dev/null +++ b/tests/test_codewars/test_snail.py @@ -0,0 +1,10 @@ +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]