arithmetic.py

def add(a, b):
    return a + b

def subt(a, b):
    return a - b

def mult(a, b):
    return a * b

def div(a, b):
    if b == 0:
        raise ValueError("0으로 나눌 수 없습니다.")
    return a / b

main.py

from arithmetic import *

add_3_4 = add(3, 4)
assert add_3_4 == 7, "합이 7이 아닙니다."
assert add_3_4 == 8, "합이 8이 아닙니다."

• assert 연산자 : 뒤에 오는 조건이 참이 아닐 시 AssertionError 발생
  • 조건문 뒤의 메시지가 함께 전달됨
• 일반적인 프로그램 코드에는 쓰이지 않음 : 테스트용 코드에 사용

test_arithmetic.py

• 파일명은 test_... 또는 ..._test.py 로 지을 것
• 함수명은 test_... 로 지을 것

from arithmetic import *

def test_add():
    assert add(2, 4) == 6
    assert add(3, 4) == 7
    # assert add(3, 4) == 8

def test_mult():
    assert mult(2, 4) == 8
    assert mult(3, 4) == 12
    # assert mult(3, 4) == 15

• pytest 설치 후 각 test_... 함수 왼쪽에 나타나는 실행버튼 확인
  • 함수명 바꿔보기
• 오류 코드 활성화하고 실행해 볼 것

pytest test.py