[python] 함수의 테스트 코드 작성

26 Dec 2023

python

함수를 작성한 후에는 테스트 코드를 작성하여 함수가 예상대로 작동하는지 확인해야 합니다. Python에서는 unittest 모듈을 사용하여 테스트를 쉽게 작성할 수 있습니다.

단위 테스트(Unit Test) 작성

import unittest

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

class TestAddFunction(unittest.TestCase):
    def test_add_positive_numbers(self):
        self.assertEqual(add(3, 5), 8)

    def test_add_negative_numbers(self):
        self.assertEqual(add(-3, -5), -8)

    def test_add_mixed_numbers(self):
        self.assertEqual(add(3, -5), -2)

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

위 예제는 add 함수를 테스트하는 단위 테스트 코드입니다. unittest.TestCase를 상속받은 클래스를 만들고 테스트 메서드를 정의하여 테스트를 수행합니다.

실행 결과

테스트를 실행하면 다음과 같은 결과가 출력됩니다.

...
----------------------------------------------------------------------
Ran 3 tests in 0.001s

OK

위와 같은 결과가 나오면 모든 테스트가 성공한 것입니다.

결론

함수를 작성한 후에는 단위 테스트 코드를 작성하여 함수의 동작을 확인하고, 예기치 않은 오류를 방지할 수 있습니다.

더 많은 정보는 Python 공식 문서에서 확인할 수 있습니다.