[PYTHON/PYTEST] assert문 : 단위 테스트하기

■ assert문을 사용해 단위 테스트하는 방법을 보여준다.

▶ math_helper.py


def add(a : int, b : int) -> int:
    return a + b

def subtract(a : int, b : int) -> int:
    return a - b

def multiply(a : int, b : int) -> int:
    return a * b

def divide(a : int, b : int) -> int:
    return a // b

def add(a : int, b : int) -> int:

return a + b

def subtract(a : int, b : int) -> int:

return a - b

def multiply(a : int, b : int) -> int:

return a * b

def divide(a : int, b : int) -> int:

return a // b

▶ test/math_helper.py


from math_helper import add, subtract, multiply, divide

def test_add() -> None:
    assert add(1, 1) == 2

def test_subtract() -> None:
    assert subtract(5, 2) == 3

def test_multiply() -> None:
    assert multiply(10, 10) == 100

def test_divide() -> None:
    assert divide(100, 25) == 4

from math_helper import add, subtract, multiply, divide

def test_add() -> None:

assert add(1, 1) == 2

def test_subtract() -> None:

assert subtract(5, 2) == 3

def test_multiply() -> None:

assert multiply(10, 10) == 100

def test_divide() -> None:

assert divide(100, 25) == 4

※ 테스트 파일명은 파일명 앞에 "test_"를 붙여야 한다.

▶ requirements.txt


exceptiongroup==1.2.1
iniconfig==2.0.0
packaging==24.0
pluggy==1.5.0
pytest==8.2.1
tomli==2.0.1

exceptiongroup==1.2.1

iniconfig==2.0.0

packaging==24.0

pluggy==1.5.0

pytest==8.2.1

tomli==2.0.1