파이썬에서는 여러가지 테스트 프레임워크를 활용하여 효율적으로 코드를 테스트할 수 있습니다. 이 글에서는 파이썬의 테스트 러너 중 하나인 unittest를 중점으로 살펴보겠습니다.

unittest 소개

unittest는 파이썬의 내장 라이브러리로서, 테스트를 위한 다양한 기능을 제공합니다. 이를 활용하여 유지보수 가능한 테스트 코드 작성과 실행, 그리고 테스트 결과를 확인할 수 있습니다.

unittest를 사용한 간단한 예제

다음은 간단한 수학 함수를 테스트하는 예제 코드입니다. 먼저, unittest 모듈을 임포트하고 TestCase를 상속하는 클래스를 생성합니다.

import unittest

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

class TestMathFunctions(unittest.TestCase):
    def test_add(self):
        self.assertEqual(add(3, 5), 8)
        self.assertEqual(add(-1, 1), 0)

if __name__ == '__main__':
    unittest.main()

이 예제에서는 add 함수를 테스트하기 위해 TestMathFunctions 클래스를 생성하고, test_add 메서드 내에서 다양한 입력값에 대한 기대값을 설정하고 테스트합니다.

unittest 실행 및 결과 확인

위의 예제 코드를 test_math_functions.py로 저장한 뒤, 터미널에서 해당 파일을 실행하면 테스트가 실행되고 결과가 출력됩니다.

$ python test_math_functions.py

결과는 다음과 같이 테스트가 성공했는지 혹은 실패했는지 여부 및 상세 정보를 보여줍니다.

참고 자료

• Python 공식 unittest 문서

이제 여러분도 unittest를 활용하여 파이썬 코드를 효과적으로 테스트해볼 수 있을 것입니다.