파이썬을 이용한 테스트 자동화

Python

DevOps

파이썬을 이용한 테스트 자동화

Author

gabriel yang

Published

October 1, 2024

소프트웨어 개발에서 테스트는 필수적인 과정입니다. 품질 높은 소프트웨어를 만들기 위해서는 코드가 예상대로 동작하는지 확인하는 테스트가 필요합니다. 하지만 코드가 커지고 복잡해질수록 수동으로 테스트를 실행하는 것은 비효율적입니다. 이때 테스트 자동화가 중요한 역할을 합니다.

파이썬은 unittest, pytest, doctest 등의 강력한 테스트 도구를 제공하여 테스트 자동화를 쉽게 구현할 수 있습니다. 이 글에서는 파이썬을 이용해 테스트 자동화를 구현하는 방법을 단계별로 알아보겠습니다.

1. 테스트 자동화란?

테스트 자동화는 코드가 원하는 대로 작동하는지 검증하는 과정을 자동으로 실행하는 것을 의미합니다. 수동 테스트와 달리, 테스트 자동화를 통해 개발자는 코드의 변경 사항이 다른 부분에 미치는 영향을 자동으로 빠르게 확인할 수 있습니다. 이는 더 안정적인 코드를 작성하고 유지보수하는 데 큰 도움이 됩니다.

2. 파이썬에서 테스트 자동화의 장점

빠르고 효율적인 테스트: 코드의 작은 수정 사항을 빠르게 검증할 수 있습니다.
버그 예방: 지속적인 테스트 실행을 통해 코드의 안정성을 유지하고 버그를 초기에 발견할 수 있습니다.
문서화의 역할: 테스트 코드는 특정 기능의 사용법을 문서화하는 데에도 유용합니다.
팀 협업 개선: 여러 개발자가 협업할 때, 테스트 자동화는 코드의 품질을 유지하며 서로의 코드가 잘 동작하는지 검증하는 데 도움을 줍니다.

3. 파이썬 테스트 도구들

파이썬은 여러 테스트 도구를 제공하며, 이 중에서 가장 많이 사용되는 몇 가지를 살펴보겠습니다.

1) unittest

unittest는 파이썬 내장 모듈로, 기본적인 단위 테스트를 작성할 수 있는 도구입니다. JUnit과 비슷한 방식으로 동작하며, 클래스와 메서드를 통해 테스트를 구성합니다.

2) pytest

pytest는 파이썬에서 가장 인기 있는 테스트 프레임워크로, 간결하면서도 강력한 기능을 제공합니다. unittest에 비해 코드가 더 직관적이며, 다양한 플러그인을 통해 확장할 수 있습니다.

3) doctest

doctest는 파이썬 코드의 주석에 적힌 예제를 테스트하는 방식입니다. 간단한 코드나 문서화가 필요한 경우 매우 유용합니다.

4. unittest로 테스트 작성하기

파이썬 표준 라이브러리인 unittest 모듈을 사용하여 간단한 테스트를 작성하는 방법을 알아보겠습니다.

1) `unittest` 기본 테스트 구조

import unittest

# 테스트할 함수 정의
def add(x, y):
    return x + y

# 테스트 케이스 정의
class TestAddFunction(unittest.TestCase):

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

    def test_add_floats(self):
        self.assertEqual(add(2.5, 3.5), 6.0)

# 테스트 실행
if __name__ == '__main__':
    unittest.main()

TestAddFunction: unittest.TestCase를 상속받아 테스트 케이스를 정의합니다.
test_add_integers: 정수 값을 더하는 테스트.
test_add_floats: 실수 값을 더하는 테스트.

이 코드를 실행하면 각 테스트 케이스가 자동으로 실행되어 결과를 출력합니다.

2) 테스트 실행 결과

..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

모든 테스트가 성공적으로 통과하면 위와 같이 출력됩니다.

5. pytest로 테스트 작성하기

pytest는 unittest보다 더 간결한 문법으로 테스트를 작성할 수 있는 도구입니다. 이제 동일한 테스트를 pytest로 작성해보겠습니다.

1) `pytest` 설치

pytest는 파이썬 표준 라이브러리가 아니므로, 먼저 설치해야 합니다.

pip install pytest

2) `pytest`를 이용한 테스트 작성

# 테스트할 함수 정의
def add(x, y):
    return x + y

# pytest로 테스트 함수 작성
def test_add_integers():
    assert add(2, 3) == 5

def test_add_floats():
    assert add(2.5, 3.5) == 6.0

pytest는 테스트 케이스를 클래스가 아닌 함수로 작성할 수 있으며, assert를 사용해 조건을 검증합니다.

3) 테스트 실행

pytest는 단순히 터미널에서 pytest 명령어를 실행하는 것만으로도 프로젝트 내의 모든 테스트를 자동으로 찾아 실행합니다.

pytest

4) pytest 실행 결과

============================= test session starts ==============================
collected 2 items

test_sample.py ..                                                       [100%]

============================== 2 passed in 0.01s ===============================

pytest는 성공한 테스트는 .으로, 실패한 테스트는 F로 표시하여 직관적으로 결과를 보여줍니다.

6. 테스트 커버리지 측정

테스트 커버리지(coverage)는 코드의 어느 부분이 테스트되었는지를 측정하는 도구입니다. pytest와 함께 pytest-cov 플러그인을 사용하여 테스트 커버리지를 확인할 수 있습니다.

1) `pytest-cov` 설치

pip install pytest-cov

2) 테스트 커버리지 실행

pytest --cov=<테스트할 모듈명>

3) 테스트 커버리지 결과

Name              Stmts   Miss  Cover
-------------------------------------
sample.py             4      0   100%

테스트가 전체 코드의 몇 퍼센트를 검증하고 있는지 쉽게 확인할 수 있습니다.

7. CI/CD와 테스트 자동화

테스트 자동화는 단독으로도 강력하지만, CI/CD(Continuous Integration/Continuous Deployment) 환경에서 더욱 큰 역할을 합니다. Jenkins, Travis CI, GitHub Actions와 같은 도구를 사용해 코드가 변경될 때마다 자동으로 테스트를 실행하고, 코드가 정상적으로 동작하는지 확인하는 자동화된 프로세스를 구축할 수 있습니다.