먼저, 테스트하려는 GPU 가속화 코드의 예제를 작성합니다. 예를 들어, 다음과 같이 간단한 GPU 가속화 함수인 add_gpu(a, b)를 가정해봅시다.

def add_gpu(a, b):
    """
    GPU 가속화된 두 수를 더하는 함수

    Parameters:
        a (int): 첫 번째 정수
        b (int): 두 번째 정수

    Returns:
        int: 두 수의 합

    Examples:
        >>> add_gpu(2, 3)
        5
        >>> add_gpu(0, 0)
        0
        >>> add_gpu(-5, 5)
        0
    """
    # 실제 GPU 가속화 코드 작성
    return a + b

그런 다음, 이 함수의 예제를 포함한 문서화 문자열(docstring)을 작성합니다. 예제는 >>>로 시작하는 입력과 해당 입력에 대한 기대되는 출력으로 구성됩니다.

이제, 테스트를 실행하기 위해 다음과 같이 doctest 모듈을 사용해야 합니다.

import doctest

if __name__ == "__main__":
    doctest.testmod()

if __name__ == "__main__":은 스크립트가 직접 실행될 때만 테스트를 실행하고, 모듈로 임포트되었을 때는 테스트를 실행하지 않도록 보장하는 역할을 합니다.

위의 예제를 포함한 파일을 저장한 뒤, 해당 파일을 실행하면 doctest가 GPU 가속화 코드의 예제를 실행하고 결과를 출력합니다. 모든 예제가 통과되면 테스트가 성공적으로 완료되었음을 알 수 있습니다.

doctest 모듈은 테스트 케이스의 일부로서 코드 예제를 포함하므로, 코드 변경 사항이 생길 때마다 테스트를 실행하여 예제와 코드 사이의 일관성을 유지할 수 있습니다.

이렇게 사용하면 doctest를 통해 GPU 가속화 코드를 효과적으로 테스트할 수 있습니다.