def test_simple_reverse_top_two(self): stack = Stack() stack.add(1) stack.add(2) reverse_top_two(stack) self.assertEqual(stack.remove(), 1) self.assertEqual(stack.remove(), 2) self.assertTrue(stack.is_empty())
def test_simple_reverse_top_two(): stack = Stack() stack.push(1) stack.push(2) reverse_top_two(stack) assert stack.pop() == 1 assert stack.pop() == 2 assert stack.is_empty()
def test_more_than_two_number(self): stack = Stack() stack.add(1) stack.add(2) stack.add(3) stack.add(4) stack.add(5) reverse_top_two(stack) self.assertEqual(stack.remove(), 4) self.assertEqual(stack.remove(), 5) self.assertFalse(stack.is_empty())
def test_more_than_two_char(self): stack = Stack() stack.add('a') stack.add('b') stack.add('c') stack.add('d') stack.add('e') reverse_top_two(stack) self.assertEqual(stack.remove(), 'd') self.assertEqual(stack.remove(), 'e') self.assertEqual(stack.remove(), 'c') self.assertEqual(stack.remove(), 'b') self.assertEqual(stack.remove(), 'a') self.assertTrue(stack.is_empty())