[Python] PyTest - parametrize 사용해보기

특정 함수를 여러번 사용 실행하는데, 입력값만 변경해서 반복적으로 호출해야되는 상황이 생기면

중복으로 테스트 케이스를 작성하는 것이 비효율적일 때가 있다.

 

이때는 @pytest.mark.parametrize parametrize 데코레이터를 사용하면 된다.

 

예를 들어 구구단 코드를 작성해본다.

import pytest

# 구구단 6 단
@pytest.mark.parametrize("num, output",
                         [(1,6),
                          (2,12),
                          (3,18),
                          (4,24)])
def test_multiplication_6(num, output):
   assert num * 6 == output

 

실행한 결과는 아래와 같다.

 

구구단 처럼 숫자가 아니어도, 특정 텍스트가 포함되어 있는지 등을 여러번 확인해야할 때

유용하게 사용할 수 있는 데코레이터이다.