[python] assert문을 사용한 테스트 결과 확인 방법

Python에서는 assert문을 사용하여 조건이 True인지 확인하고, 그렇지 않다면 AssertionError를 발생시킵니다. 이를 테스트 작성 및 디버깅에 활용할 수 있습니다.

assert문의 기본 구조

assert 조건, "예외 메시지"

위의 구조에서 조건이 False이면 AssertionError가 발생하며, 연관된 예외 메시지가 함께 출력됩니다.

테스트 케이스에서의 활용

보통 테스트 케이스에서 assert문을 활용하여 예상되는 결과를 검증합니다. 예를 들어, 함수의 반환 값이 예상한 값과 일치하는지 확인할 수 있습니다.

def add(a, b):
    return a + b

result = add(3, 5)
assert result == 8, "add 함수의 결과가 올바르지 않습니다"

위 예제에서, add 함수가 3과 5를 더한 결과가 8이 아니라면 AssertionError가 발생하며 메시지가 출력됩니다.

디버깅에서의 활용

assert문은 또한 코드 실행 중에 특정 시점의 조건을 검증하는 데에도 사용될 수 있습니다.

# some complex calculations
assert condition, "특정 시점의 조건이 충족되지 않았습니다"

위 예제에서, condition이 False인 경우 AssertionError가 발생하여 해당 시점의 조건을 디버깅할 수 있습니다.

결론

assert문은 테스트 케이스 작성 및 코드 디버깅에 유용하게 사용될 수 있는 Python의 기능입니다. 잘 활용하면 코드의 신뢰성을 높일 수 있습니다.


참고 자료: