Python unittest 완전 가이드 | 기본부터 응용까지 철저히 해설

1. Python unittest란?

유닛 테스트의 중요성

2. unittest의 기본적인 사용법

기본적인 테스트 예시</h3음 코드는 add(a, b) 함수를 테스트하는 간단한 예시입니다。

import unittest

# 테스트 대상 코드
def add(a, b):
    return a + b

# 테스트 클래스
class TestAddFunction(unittest.TestCase):

    def test_add_integers(self):
        result = add(2, 3)
        self.assertEqual(result, 5)

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

테스트 확장

def test_add_floats(self):
    result = add(2.5, 3.5)
    self.assertAlmostEqual(result, 6.0, places=2)

def test_add_strings(self):
    result = add("Hello, ", "World!")
    self.assertEqual(result, "Hello, World!")

3. setUp()와 tearDown() 사용법

setUp() 예시

def setUp(self):
    self.temp_value = 42

tearDown() 예시

def tearDown(self):
    self.temp_value = None

4. 모크를 사용한 의존 관계 테스트

모크 예시

from unittest.mock import patch

class TestAddFunction(unittest.TestCase):

    @patch('my_module.time_consuming_function')
    def test_add_with_mock(self, mock_func):
        mock_func.return_value = 0
        result = add(2, 3)
        self.assertEqual(result, 5)

5. 예외 처리와 커스텀 어설션

예외 처리 테스트

def test_divide_by_zero(self):
    with self.assertRaises(ZeroDivisionError):
        divide(1, 0)

커스텀 어설션 만들기

def assertIsPositive(self, value):
    self.assertTrue(value > 0, f'{value} is not positive')

6. unittest의 테스트 디스커버리 기능

테스트 디스커버리 사용 방법

python -m unittest discover

python -m unittest discover -s tests -p "test_*.py"

7. unittest를 사용한 성능 향상 팁

파일 I/O 최적화

from io import StringIO

class TestFileOperations(unittest.TestCase):

    def test_write_to_memory(self):
        output = StringIO()
        output.write('Hello, World!')
        self.assertEqual(output.getvalue(), 'Hello, World!')

모크 사용하기

from unittest.mock import MagicMock

class TestApiCall(unittest.TestCase):

    def test_api_response(self):
        mock_api = MagicMock(return_value={'status': 'success'})
        response = mock_api()
        self.assertEqual(response['status'], 'success')

8. 요약 및 다음 단계

주요 포인트 요약

• 기본적인 사용법: unittest.TestCase를 상속하고, 어설션 메서드를 활용하여 테스트를 작성합니다。
• setUp() / tearDown(): 테스트 전후에 공통 처리를 모아 관리함으로써 코드 재사용성과 가독성을 향상시킵니다。
• 모크 활용: 외부 리소스에 의존하지 않고 기능을 테스트할 수 있어 테스트 효율이 크게 향상됩니다。
• 테스트 디스커버리: 대규모 프로젝트에서 테스트 관리를 쉽게 해주는 편리한 기능입니다。
• 성능 향상 기술: 메모리 상에서 처리하거나 모크를 활용함으로써 테스트 실행 시간을 단축할 수 있습니다。

다음 단계