Пример #1
0
def test_minstack():
    minStack = MinStack()
    minStack.push(-2)
    minStack.push(0)
    minStack.push(-3)
    assert minStack.getMin() == -3
    minStack.pop()
    assert minStack.top() == 0
    assert minStack.getMin() == -2
Пример #2
0
  def test_top(self):
    # Given
    # [1, 2, 3, 4, 5, 6, 7]
    minStack = MinStack()
    minStack.push(1)
    minStack.push(2)
    minStack.push(3)
    minStack.push(4)
    minStack.push(5)
    minStack.push(6)
    minStack.push(7)

    expected_output = 7
    self.assertEqual(minStack.top(), expected_output)
Пример #3
0
class TestMinStack(unittest.TestCase):
    def setUp(self):
        self.stack = MinStack()
        self.stack.push(2)
        self.stack.push(1)

    def test_push(self):
        self.stack.push(3)
        self.assertEqual(self.stack.stack[-1], 3)

    def test_pop(self):
        self.stack.pop()
        self.assertEqual(self.stack.stack[-1], 2)

    def test_top(self):
        self.assertEqual(self.stack.top(), 1)

    def test_getMin(self):
        self.assertEqual(self.stack.getMin(), 1)
        self.stack.push(-1000)
        self.assertEqual(self.stack.getMin(), -1000)