[Python] pytest로 exception assert하기

pytest를 설치는 하였고 이제 Exception을 처리하는 방법에 대해서 알아보자


환경 및 선수조건


기본적인 사용법

  • Exception의 오류를 잡기 위해서 pytest에서는 pytest.raises([Exception])이라는 함수를 통해서 실행시킵니다. 바로 코드를 보도록 하겠습니다.
  • 아주 간단한 코드로 with문을 참고하시면 ()안에 있는 Exception이 발생하는지를 확인하는 코드입니다. with문 안에 1/0의 코드를 통해서 해당 Exception을 발생시킵니다.

test_zero_division.py

import pytest


def test_zero_division():
    # with문 괄호에 있는 Exception이 해당 block에서 일어나는지 확인하는 코드
    with pytest.raises(ZeroDivisionError):
        1 / 0


이제 코드를 작성하였으니 실행시켜봅시다.

Run Test


참고자료