Python 유닛 테스트 완전 가이드 (pytest 중심)

자, 파이썬 유닛 테스트 이야기, 시작해볼까요? 솔직히 말씀드리면, 저도 처음엔 파이썬으로 개발하면서 테스트 코드 작성이 얼마나 중요한지 잘 몰랐어요. 그냥 돌아가기만 하면 되는 거 아닌가? 싶었죠. 근데 프로젝트 규모가 커지고 기능이 복잡해지니까… 끔찍한 버그들이 쏟아지는 거예요. 그때부터 테스트의 중요성을 절실히 깨달았습니다. 그래서 오늘은 제가 pytest라는 훌륭한 도구를 통해 얻은 경험을 여러분과 나누려고 합니다!

pytest는 파이썬 유닛 테스트를 위한 프레임워크인데요, 쉽게 말해 우리가 짠 코드 조각(함수나 클래스)이 제대로 작동하는지 확인하는 도구라고 생각하시면 됩니다. 마치 레고 블록 하나하나가 제대로 맞는지 확인하는 것과 같아요. 이게 왜 중요하냐구요? 작은 오류 하나 때문에 전체 시스템이 무너지는 걸 막아주거든요. 게다가 나중에 코드를 수정하거나 기능을 추가할 때도 훨씬 안전하고 편해집니다. 미리 테스트 해 놓으면, 내가 뭘 고쳤는데 엉뚱한 곳이 망가졌는지 바로 알 수 있으니까요.

자, 그럼 유닛 테스트의 기본 개념부터 살펴볼까요? 우선 ‘유닛’이란 각각 독립적으로 작동하는 코드의 최소 단위를 말해요. 예를 들어, 숫자 두 개를 더하는 함수라면 그 함수 자체가 하나의 유닛이 되겠죠. pytest를 이용하면 이런 유닛들을 하나씩 테스트할 수 있습니다. 테스트 함수는 test_로 시작하는 이름을 붙여주면 되는데, 예를 들어 test_add() 라는 함수가 있다면, add() 함수를 테스트하는 함수라는 걸 쉽게 알 수 있겠죠? 그리고 assert 문을 이용해 예상 결과와 실제 결과를 비교합니다. 결과가 다르면 테스트가 실패하고, 그 부분을 고쳐야 한다는 신호를 받는 거죠.

코드 예제를 보여드릴게요. 아래 코드는 add() 함수를 테스트하는 예시입니다.

import pytest

def add(x, y):
  return x + y

def test_add():
  assert add(2, 3) == 5
  assert add(-1, 1) == 0
  assert add(0, 0) == 0

간단하죠? 이렇게 여러 개의 assert 문을 사용해서 다양한 상황을 테스트할 수 있습니다. 물론, 단순히 숫자만 더하는 것뿐만 아니라, 문자열을 처리하거나 파일을 읽고 쓰는 등 더 복잡한 함수도 같은 방식으로 테스트할 수 있어요.

그리고 pytest에는 @pytest.fixture 라는 아주 유용한 기능이 있습니다. 이건 테스트에 필요한 데이터나 리소스를 미리 준비해주는 역할을 해요. 마치 요리할 때 재료를 미리 손질해 놓는 것과 비슷하죠. 예를 들어, 테스트에 필요한 데이터 목록을 my_data라는 fixture로 만들어 놓으면, 각 테스트 함수에서 이 데이터를 편리하게 사용할 수 있습니다. 또, @pytest.mark.parametrize 데코레이터를 이용하면 여러 입력값을 한꺼번에 테스트할 수도 있어요. 이렇게 하면 테스트 코드를 훨씬 간결하고 효율적으로 작성할 수 있습니다.

하지만, 테스트 커버리지도 중요합니다. 테스트 커버리지는 전체 코드 중 얼마나 많은 부분이 테스트되었는지를 나타내는 지표인데요, 100% 커버리지가 항상 좋은 건 아니지만, 중요한 기능은 꼼꼼하게 테스트하는 게 좋겠죠? pytest-cov 플러그인을 사용하면 커버리지를 쉽게 측정할 수 있습니다. 그리고 테스트는 각각 독립적으로 실행되어야 한다는 점, 그리고 테스트 함수의 이름은 테스트 내용을 명확하게 나타내도록 짓는 것도 중요한 부분입니다.

마지막으로, 꾸준히 테스트하는 습관을 들이는 게 정말 중요해요. 처음에는 귀찮을 수 있지만, 나중에 큰 도움이 된다는 걸 여러분도 곧 깨닫게 될 거예요. pytest를 활용해서 체계적인 유닛 테스트를 하면 코드 품질을 높이고, 장기적으로 훨씬 효율적인 개발을 할 수 있답니다. pytest의 다양한 기능들을 잘 활용해서 여러분만의 효과적인 테스트 환경을 만들어 보세요! 어려운 점이 있으면 언제든지 질문해주세요! 함께 고민해 봐요!