def test_push_and_pop(self): stack = StackWithMin() stack.push(1) stack.push(2) stack.push(3) self.assertEqual(stack.pop(), 3) self.assertEqual(stack.pop(), 2)
def test_min(self): stack = StackWithMin() stack.push(3) stack.push(2) stack.push(1) self.assertEqual(stack.min(), 1) stack.pop() self.assertEqual(stack.min(), 2) stack.pop() self.assertEqual(stack.min(), 3)