파이썬 테스트 코드(pytest) 입문

파이썬 코드를 작성하는 개발자라면 누구나 효율적인 테스트의 중요성을 알고 있을 겁니다.

하지만 테스트 코드를 작성하는 과정이 번거롭고 어렵게 느껴질 때도 많죠.

pytest는 이러한 고민을 해결하고 파이썬 테스트를 더욱 쉽고 강력하게 만들어주는 훌륭한 도구입니다.

이 글에서는 pytest를 활용한 파이썬 테스트 입문 방법을 소개합니다.

 

pytest 설치 및 기본 실행

 

pytest를 사용하기 위해서는 먼저 설치가 필요합니다.

간단하게 `pip install pytest` 명령어를 통해 설치할 수 있습니다.

설치가 완료되면, 현재 디렉토리에서 `test_*.py` 또는 `*_test.py` 형식의 모든 파일을 자동으로 찾아 테스트를 실행하는 `pytest` 명령어를 사용할 수 있습니다.

특정 파일만 테스트하고 싶다면 `pytest test_file.py`와 같이 파일명을 지정할 수 있고, 특정 경로 내의 테스트를 실행하려면 `pytest testing/`과 같이 경로를 지정할 수도 있습니다.

 

테스트 대상 지정 및 제외

 

pytest는 다양한 방법으로 테스트 대상을 지정할 수 있습니다.

`-k` 옵션을 사용하면 특정 키워드를 포함하는 테스트만 실행할 수 있습니다.

예를 들어, `pytest -k "login"`은 "login"이라는 단어가 포함된 테스트 함수만 실행합니다.

반대로 특정 경로를 테스트에서 제외하고 싶을 수도 있습니다.

이러한 경우에는 pytest 설정 파일(pytest.ini 또는 pyproject.toml)을 통해 제외할 경로를 지정할 수 있습니다.

이렇게 하면 불필요한 테스트 실행 시간을 줄이고 효율적인 테스트 환경을 구축할 수 있습니다.

 

Fixture를 활용한 테스트 환경 설정

 

pytest의 강력한 기능 중 하나는 fixture입니다.

Fixture는 테스트 실행 전 필요한 객체를 생성하거나 특정 상태를 설정하는 데 사용됩니다.

예를 들어, 데이터베이스 연결, API 클라이언트 생성, 또는 복잡한 객체 초기화 등을 fixture를 통해 처리할 수 있습니다.

`@pytest.fixture` 데코레이터를 사용하여 fixture를 정의하고, 테스트 함수에서 인자로 받아 사용할 수 있습니다.

fixture는 테스트 코드의 중복을 줄이고 가독성을 높이는 데 큰 도움을 줍니다.

 

테스트 스킵 및 병렬 실행

 

때로는 특정 테스트를 건너뛰어야 할 경우가 있습니다.

예를 들어, 아직 구현되지 않은 기능에 대한 테스트나 특정 환경에서만 실행 가능한 테스트가 있을 수 있습니다.

pytest에서는 `@pytest.mark.skip` 데코레이터를 사용하여 테스트를 스킵할 수 있습니다.

또한, pytest-xdist 플러그인을 사용하면 테스트를 병렬로 실행하여 전체 테스트 시간을 단축할 수 있습니다.

병렬 실행은 특히 테스트 스위트가 큰 프로젝트에서 유용합니다.

 

테스트 실행 시간 측정 및 teardown 코드 실행

 

pytest는 각 테스트의 실행 시간을 측정하는 기능을 제공합니다.

이를 통해 성능 병목 지점을 파악하고 코드 최적화에 활용할 수 있습니다.

또한, fixture에서 `yield` 키워드를 사용하면 테스트 실행 후 teardown 코드를 실행할 수 있습니다.

Teardown 코드는 테스트 과정에서 생성된 리소스를 정리하거나 데이터베이스 상태를 초기화하는 데 사용됩니다.

이를 통해 테스트 환경을 깨끗하게 유지하고 다음 테스트에 영향을 미치지 않도록 할 수 있습니다.

 

pytest는 파이썬 테스트를 위한 강력하고 유연한 도구입니다.

다양한 옵션과 플러그인을 통해 테스트 환경을 사용자에 맞게 구성할 수 있으며, fixture 기능을 통해 테스트 코드의 효율성과 가독성을 높일 수 있습니다.

최근 파이썬 프로젝트에서 pytest를 적극적으로 활용하여 효율적인 테스트 환경을 구축하고 코드 품질을 향상시키는 것은 이제 선택이 아닌 필수입니다.

지금 바로 pytest를 시작하여 여러분의 파이썬 개발 경험을 한 단계 업그레이드해 보세요.