def test_1(self): """Shuffle works""" deck_of_cards = Solution([num for num in range(52)]) deck_of_cards2 = Solution([num for num in range(52)]) result_shuffle = deck_of_cards.shuffle() result_shuffle2 = deck_of_cards2.shuffle() self.assertEqual(result_shuffle == result_shuffle2, False)
def test_2(self): """Reset works""" array_100_nums = [num for num in range(0, 100)] obj = Solution(list(array_100_nums)) obj.shuffle() self.assertEqual(obj.reset(), array_100_nums) obj.shuffle() self.assertEqual(obj.reset(), array_100_nums)
from main import Solution s = Solution() nums = [2, 5, 1, 3, 4, 7] n = 3 print("case 1 output:", s.shuffle(nums, n)) nums = [1, 2, 3, 4, 4, 3, 2, 1] n = 4 print("case 2 output:", s.shuffle(nums, n))