[python] setUp()과 tearDown() 메소드의 활용

15 Dec 2023

python

unittest 모듈은 Python의 표준 라이브러리 중 하나로, 테스트 케이스를 생성하고 실행하는 데 사용된다.

unittest.TestCase 클래스는 테스트 메소드를 실행하기 전에 setUp() 메소드를 호출하여 테스트 환경을 설정하고, 테스트 메소드 실행 후에 tearDown() 메소드를 호출하여 환경을 정리한다. 이러한 메소드들을 이용하면 반복적인 초기화 및 정리 작업을 간편하게 수행할 수 있다.

아래는 setUp()과 tearDown() 메소드의 활용 예시이다.

import unittest

class MyTestCase(unittest.TestCase):
    def setUp(self):
        # 테스트 환경 설정
        self.file = open('test_file.txt', 'w')
    
    def tearDown(self):
        # 테스트 환경 정리
        self.file.close()
    
    def test_file_write(self):
        # 파일에 쓰기 작업을 수행하는 테스트 메소드
        self.file.write('Hello, world!')
        self.assertEqual(self.file.name, 'test_file.txt')
    
    def test_file_read(self):
        # 파일 읽기 작업을 수행하는 테스트 메소드
        self.file.write('Hello, world!')
        self.file.seek(0)
        content = self.file.read()
        self.assertEqual(content, 'Hello, world!')

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

위와 같이 setUp() 메소드를 이용하여 테스트 전 사전 작업을 수행하고, tearDown() 메소드를 이용하여 테스트 후 후속 작업을 수행하는 것이, 테스트 코드를 보다 체계적으로 관리하고 유지보수하기 쉽게 해준다.

이와 같은 단위 테스트를 통해 코드를 작성하면, 코드의 신뢰성을 높일 수 있고, 버그를 줄이고 유지보수성을 향상시킬 수 있다.

더 많은 정보를 원하시면 Python 공식 문서를 참고하시기 바랍니다.