학습 목표
이번 강의에서는 테스트 코드를 작성하여 코드의 품질을 높이는 방법을 배웁니다. 다양한 형태의 테스트를 구현하여 코드의 신뢰성을 강화하고, 버그를 조기에 발견하는 기술을 익힐 것입니다.
개념 설명
테스트 코드는 소프트웨어의 기능이 기대대로 작동하는지 검증하기 위해 작성하는 코드입니다. 테스트 코드의 주요 목적은 코드에 대한 신뢰성을 높이고, 새로운 변경 사항이 기존의 기능에 부정적인 영향을 주지 않도록 하는 것입니다. 파이썬에서는 unittest 모듈을 사용하여 테스트를 작성할 수 있습니다. 주요 요소는 다음과 같습니다:
- 테스트 케이스(Test Case): 특정 기능이나 메서드의 동작을 검증하는 클래스입니다.
unittest.TestCase를 상속하여 작성합니다. - 어설션(Assertion): 특정 조건이 참인지 검사하는 메서드입니다. 예를 들어,
assertEqual(),assertTrue()등이 있습니다. - 테스트 스위트(Test Suite): 여러 테스트 케이스를 묶어 한 번에 실행할 수 있는 구조입니다.
- 테스트 러너(Test Runner): 테스트를 실행하고 결과를 보고하는 역할을 합니다.
예제 코드
# 테스트할 함수
def multiply(a, b):
return a * b
# 유닛 테스트 정의
import unittest
class TestMathFunctions(unittest.TestCase):
def test_multiply(self):
self.assertEqual(multiply(2, 3), 6) # 2 * 3 = 6
self.assertEqual(multiply(0, 5), 0) # 0 * 5 = 0
self.assertEqual(multiply(-1, 1), -1) # -1 * 1 = -1
self.assertNotEqual(multiply(5, 5), 20) # 5 * 5는 20이 아님
# 테스트 실행
if __name__ == "__main__":
unittest.main() # 모든 테스트 케이스 실행코드 설명
이 코드는 unittest 모듈을 사용하여 간단한 곱셈 함수를 테스트하는 예제입니다.
- 먼저,
multiply함수를 정의합니다. TestMathFunctions클래스는unittest.TestCase를 상속받아 곱셈 함수에 대한 테스트 케이스를 정의합니다.test_multiply메서드는 여러 가지 어설션을 사용하여 곱셈 함수의 출력을 검증합니다.- 메인 블록에서
unittest.main()을 호출하여 모든 테스트 케이스를 실행합니다.
연습 문제
1. 덧셈을 수행하는 함수를 작성하고, 이 함수를 테스트하는 유닛 테스트를 작성해 보세요.
2. 리스트의 가장 큰 요소를 반환하는 함수를 작성하고, 다양한 테스트 케이스를 작성하여 검증해 보세요.
연습 문제 해답
# 1번 문제: 덧셈 함수
def add(a, b):
return a + b
# 덧셈 함수 테스트
class TestAddFunction(unittest.TestCase):
def test_add(self):
self.assertEqual(add(2, 3), 5) # 2 + 3 = 5
self.assertEqual(add(-1, 1), 0) # -1 + 1 = 0
self.assertEqual(add(0, 0), 0) # 0 + 0 = 0
self.assertNotEqual(add(5, 7), 10) # 5 + 7는 10이 아님
# 2번 문제: 리스트의 최대값 반환하는 함수
def max_in_list(lst):
return max(lst)
# 최대값 반환하는 함수 테스트
class TestMaxInListFunction(unittest.TestCase):
def test_max_in_list(self):
self.assertEqual(max_in_list([1, 2, 3, 4]), 4) # 최대값 4
self.assertEqual(max_in_list([-10, -5, 0]), 0) # 최대값 0
self.assertEqual(max_in_list([5]), 5) # 한 요소인 경우
if __name__ == "__main__":
unittest.main() # 모든 테스트 케이스 실행
0 댓글