[Python/파이썬]assert 문 활용 방법

1. 기본적인 assert 문 사용

assert 문은 주어진 조건이 True가 아니면 AssertionError를 발생시킵니다. 이를 사용하여 프로그램의 상태나 조건을 검증할 수 있습니다.

x = 5
assert x == 5

2. AssertionError 메시지 지정

assert 문에는 선택적으로 AssertionError 메시지를 지정할 수 있습니다. 이를 통해 조건이 충족되지 않았을 때 추가적인 정보를 제공할 수 있습니다.

y = 10
assert y == 5, "y의 값이 5가 아닙니다."

3. assert 문을 함수 내부에서 사용

assert 문을 함수 내부에서 사용하여 함수의 입력 값이나 결과를 검증할 수 있습니다. 이를 통해 함수의 올바른 동작을 보장할 수 있습니다.

def divide(a, b):
    assert b != 0, "0으로 나눌 수 없습니다."
    return a / b

result = divide(10, 0)

4. assert 문을 테스트에 활용

단위 테스트에서 assert 문은 특히 유용합니다. 예상한 대로 동작하지 않을 경우 AssertionError가 발생하여 문제를 신속하게 식별할 수 있습니다.

def test_addition():
    assert add(3, 5) == 8
    assert add(-1, 1) == 0

5. 디버깅에 활용

assert 문은 디버깅 시에도 유용하게 사용될 수 있습니다. 프로그램의 특정 시점에서 변수의 상태를 검증하여 잘못된 값이 사용되는지 여부를 확인할 수 있습니다.

assert len(items) > 0, "items 리스트는 비어 있습니다."