[Python] pytest.raises 사용해보기

2024.05.01 - [🎪 놀고있네/Python] - [Python] Pytest - Xfail Tests

위에 글에서 1 / 0 의 결과로 ZeroDivisionError 가 노출되었는데, 해결 방법을 찾아보자.

예외처리할때 가장 먼저 생각나는 건 try-except 이다. 일반적인 설명은

 

try-except 문:

• try-except 문은 예외가 발생할 수 있는 코드 블록을 시도하고, 예외가 발생했을 때 이를 처리합니다.
• 코드 블록에서 예외가 발생하면 프로그램의 실행이 중단되지 않고, 예외를 처리할 수 있습니다.
• try 블록 안에 있는 코드가 예외를 발생시키면, 해당 예외를 처리하는 except 블록이 실행됩니다.

 

코드로 작성해보면

import pytest

def test_divide_by_0():
    try:
        result = 1 / 0
    except ZeroDivisionError:
        print("Cannot divide by zero")

 

 

try-except 보다 간단하게 pytest 는 with과 pytest.raises 를 사용해서 예외처리를 할 수 있다! (오호)

 

파이썬에서 with 문은 파일을 열고 닫는 것과 같은 리소스 관리를 위해 일반적으로 사용되지만, pytest.raises는 pytest에서 제공하는 특수한 기능으로, 예외가 발생하는지 테스트하는 데 사용된다.

 

일반적인 with 문:

• with 문은 리소스 관리를 위해 사용됩니다. 주로 파일, 네트워크 연결, 데이터베이스 연결 등과 같은 리소스를 안전하게 사용하는 데에 활용됩니다.
• with 문을 사용하면 리소스를 사용한 후에 자동으로 정리(clean-up)됩니다. 이는 리소스 누수를 방지하고 코드를 간결하게 만듭니다.

 

def test_divide_by_0_with():
    with pytest.raises(ZeroDivisionError):
        assert 1 / 0

 

실행해보면 둘다 패쓰