def test_peek_general(items: List[int]) -> None: """Test that peek works for a large range of stack sizes.""" stack = Stack() for item in items: stack.push(item) assert peek(stack) == items[-1] assert stack.pop() == items[-1]
def test_peek_doctest() -> None: """This is the doctest given in peek.""" stack = Stack() stack.push(1) stack.push(2) assert peek(stack) == 2 assert stack.pop() == 2