아래는 assert 문을 사용한 예시 코드입니다.

def square(x):
    """
    주어진 숫자의 제곱을 반환하는 함수

    >>> assert square(2) == 4
    >>> assert square(5) == 25
    >>> assert square(0) == 0
    """
    return x ** 2

위의 예시 코드에서는 assert 문을 사용하여 square 함수의 반환값이 예상한 값과 일치하는지 확인하고 있습니다. doctest는 이러한 assert 문을 통해 예상한 결과와 실제 결과가 일치하는지 테스트하여 함수의 동작을 확인합니다.

참고로, assert 문은 테스트를 위한 코드이므로 실제 코드에서는 주의해서 사용해야 합니다. assert 문을 사용하는 경우, 문제가 발생했을 때 AssertionError가 발생하므로 문제 상황을 적절하게 처리할 수 있어야 합니다.

자세한 내용은 Python 공식 문서를 참고하시기 바랍니다.