[파이썬][리스트] 리스트 데이터를 테스트하는 방법과 예제

22 Aug 2023

python

리스트 데이터를 테스트하는 방법은 리스트에 적용되는 연산 및 기능에 대한 테스트 케이스를 작성하여 리스트의 동작을 확인하는 것입니다. 이를 위해 파이썬의 테스트 프레임워크 중 하나인 unittest를 사용할 수 있습니다. 아래는 리스트 데이터를 테스트하는 예제와 방법을 보여줍니다.

import unittest

# 리스트 데이터를 조작하는 함수 예시
def add_to_list(my_list, item):
    my_list.append(item)

def remove_from_list(my_list, item):
    my_list.remove(item)

def find_index(my_list, item):
    try:
        index = my_list.index(item)
        return index
    except ValueError:
        return -1

class TestListOperations(unittest.TestCase):
    def test_add_to_list(self):
        my_list = []
        add_to_list(my_list, 5)
        self.assertEqual(my_list, [5])

    def test_remove_from_list(self):
        my_list = [1, 2, 3, 4]
        remove_from_list(my_list, 3)
        self.assertEqual(my_list, [1, 2, 4])

    def test_find_index(self):
        my_list = ["apple", "banana", "cherry"]
        index = find_index(my_list, "banana")
        self.assertEqual(index, 1)
        index = find_index(my_list, "grape")
        self.assertEqual(index, -1)

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

위의 코드에서는 unittest를 사용하여 리스트 조작 함수에 대한 테스트 케이스를 작성했습니다. TestListOperations 클래스 안에 각 함수에 대한 테스트 메서드를 작성하고, assertEqual() 메서드를 사용하여 예상 결과와 실제 결과를 비교합니다.

테스트는 테스트 케이스마다 각 함수를 호출하고 결과를 비교하여 함수가 예상대로 동작하는지 확인합니다. 이렇게 테스트를 작성하면 코드 변경 시 기존 동작에 영향을 주지 않고 안정적으로 코드를 유지할 수 있습니다.