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())
示例#2
0
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()
示例#3
0
 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())
示例#4
0
 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())