파이썬 unittest 모듈 입문 ✅

파이썬 unittest 모듈 입문 ✅

unittest는 파이썬의 표준 단위 테스트 프레임워크입니다. 함수나 클래스의 동작을 자동으로 검증하여 코드의 품질을 높이는 데 필수적인 도구입니다.

1. 간단한 함수 테스트 작성

# calc.py
def add(a, b):
    return a + b

# test_calc.py
import unittest
from calc import add

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

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

2. `setUp()`과 `tearDown()`으로 반복 작업 처리

# test_list.py
import unittest

class TestList(unittest.TestCase):
    def setUp(self):
        self.lst = [1, 2, 3]

    def tearDown(self):
        self.lst.clear()

    def test_append(self):
        self.lst.append(4)
        self.assertIn(4, self.lst)

    def test_pop(self):
        value = self.lst.pop()
        self.assertEqual(value, 3)

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

3. 예외 발생 테스트

# test_divide.py
import unittest

def divide(a, b):
    return a / b

class TestDivide(unittest.TestCase):
    def test_zero_division(self):
        with self.assertRaises(ZeroDivisionError):
            divide(1, 0)

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

4. 테스트 실행 및 결과 확인

$ python -m unittest discover
...
----------------------------------------------------------------------
Ran 5 tests in 0.002s

OK

요약

• TestCase 클래스로 테스트 케이스 작성
• assertEqual, assertIn, assertRaises 등 다양한 검증 메서드 제공
• setUp/tearDown으로 테스트 초기화 및 정리 작업 수행
• `unittest discover`로 프로젝트 전체 테스트 자동 실행 가능