def test_cut(self): shuffler = Shuffler(10) shuffler.cut(3) self.assertEqual(shuffler.stack, [3, 4, 5, 6, 7, 8, 9, 0, 1, 2]) shuffler.reset_stack() shuffler.cut(-4) self.assertEqual(shuffler.stack, [6, 7, 8, 9, 0, 1, 2, 3, 4, 5])
def test_reset_stack(self): shuffler = Shuffler() self.assertEqual(shuffler.stack, []) shuffler.reset_stack(7) self.assertEqual(shuffler.stack, [0, 1, 2, 3, 4, 5, 6]) shuffler.stack[0] = -1 shuffler.reset_stack() self.assertEqual(shuffler.stack, [0, 1, 2, 3, 4, 5, 6])