import unittest
from lettcode.shuffleTheArr import Solution


class ShuffleTheArrTest(unittest.TestCase):
    def test_solution(self):
        solution = Solution()
        self.assertEqual(solution.shuffle([2, 5, 1, 3, 4, 7], 3), [2, 3, 5, 4, 1, 7])
        self.assertEqual(solution.shuffle([1, 2, 3, 4, 4, 3, 2, 1], 4), [1, 4, 2, 3, 3, 2, 4, 1])
        self.assertEqual(solution.shuffle([1, 1, 2, 2], 2), [1, 2, 1, 2])
