[Python] 단위 테스트: pytest

pytest는 python 환경의 대표적인 단위테스트 도구 중 하나다.

설치

우선 아래와 같이 설치하고

pip install pytest

잘 깔렸는지 확인한다.

테스트 만들어보기

다음과 같이 덧셈 함수가 있다고 가정하겠다.

저걸 테스트하려면 아래와 같이 테스트용 함수를 별도로 정의해주고, assert 키워드로 테스트 성공여부를 검증하면 된다.

test_로 시작하는 함수만 자동으로 테스트 함수로 간주된다.

그리고 pytest로 실행을 해보면

결과를 알려줄 것이다.

실패하면 이렇게 뜬다.

테스트 파일 분리하기

위에서는 단일 파일에 실제 코드와 테스트코드를 뭉쳐놨는데, 장기적으로는 분리해서 관리하는 것이 좋다.
함수명과 마찬가지로 테스트용 파일명도 test_로 시작해야 한다.

이렇게 해놓고

다음과 같이 실행 컨텍스트를 넣어서 실행해주면 그 경로 아래의 테스트파일을 전부 읽어서 처리해준다.

python -m pytest .