[파이썬] unittest의 주요 구성요소

07 Sep 2023

unittest

unittest 모듈은 파이썬에서 테스트 주도 개발을 실현할 수 있는 강력한 도구입니다. 이 모듈을 사용하여 코드의 정확성을 검증하고, 버그를 발견하고, 소스 코드를 유지보수하기 쉬운 상태로 유지할 수 있습니다. unittest의 주요 구성요소는 다음과 같습니다:

1. 테스트 케이스 (Test Case)

테스트 케이스는 unittest 모듈에서 테스트의 기본 단위입니다. 각각의 테스트 케이스는 특정 조건 및 상황에 대한 테스트를 정의하며, 테스트할 함수 또는 메서드를 호출하고 결과를 기대하는 방법을 포함합니다. unittest.TestCase를 상속받아 테스트 케이스를 작성할 수 있습니다.

import unittest

class MyTestCase(unittest.TestCase):
    def test_addition(self):
        result = 2 + 2
        self.assertEqual(result, 4)

2. 테스트 스위트 (Test Suite)

테스트 스위트는 여러 테스트 케이스를 모아서 실행할 수 있는 단위입니다. unittest.TestSuite() 클래스를 사용하여 테스트 스위트를 생성하고, addTest() 메서드를 사용하여 테스트 케이스를 추가할 수 있습니다. 또는 테스트 스위트를 직접 생성하지 않고 unittest.TestLoader를 사용하여 동적으로 테스트 케이스를 수집할 수도 있습니다.

import unittest

class MyTestCase(unittest.TestCase):
    def test_addition(self):
        result = 2 + 2
        self.assertEqual(result, 4)

class MyOtherTestCase(unittest.TestCase):
    def test_subtraction(self):
        result = 5 - 3
        self.assertEqual(result, 2)

suite = unittest.TestSuite()
suite.addTest(MyTestCase('test_addition'))
suite.addTest(MyOtherTestCase('test_subtraction'))

3. 테스트 러너 (Test Runner)

테스트 러너는 테스트를 실행하고 결과를 표시하는 역할을 합니다. unittest.TextTestRunner() 클래스를 사용하여 테스트를 터미널에 표시하고, run() 메서드를 사용하여 테스트 스위트를 실행할 수 있습니다. 또는 다른 테스트 러너를 사용하여 테스트를 실행하고 결과를 보고할 수 있습니다.

import unittest

class MyTestCase(unittest.TestCase):
    def test_addition(self):
        result = 2 + 2
        self.assertEqual(result, 4)

class MyOtherTestCase(unittest.TestCase):
    def test_subtraction(self):
        result = 5 - 3
        self.assertEqual(result, 2)

suite = unittest.TestSuite()
suite.addTest(MyTestCase('test_addition'))
suite.addTest(MyOtherTestCase('test_subtraction'))

runner = unittest.TextTestRunner()
runner.run(suite)

unittest의 주요 구성요소를 적절히 활용하면 파이썬 코드의 품질을 유지하고 개발 과정을 안정화할 수 있습니다. 테스트 코드를 작성하고 실행하는 것은 안정성 있는 소프트웨어 개발에 필수적인 과정이므로, unittest를 배우고 활용하는 것을 권장합니다.