[python] 테스트 데이터의 동적 생성하기

21 Nov 2023

python

소프트웨어 개발 과정에서 중요한 부분 중 하나는 테스트 데이터를 생성하는 것입니다. 테스트 데이터는 프로그램이 예상대로 동작하는지 확인하는 데에 필요합니다. 이번 포스트에서는 파이썬을 사용하여 테스트 데이터를 동적으로 생성하는 방법을 알아보겠습니다.

1. 랜덤 데이터 생성하기

import random

# 0부터 100 사이의 랜덤 숫자 생성
random_number = random.randint(0, 100)

# A부터 Z까지의 랜덤 문자 생성
random_character = random.choice('ABCDEFGHIJKLMNOPQRSTUVWXYZ')

# 0.0부터 1.0 사이의 랜덤 소수 생성
random_float = random.uniform(0.0, 1.0)

# 리스트에서 랜덤하게 요소 추출
random_element = random.choice(['apple', 'banana', 'orange'])

# 리스트에서 무작위로 섞기
my_list = [1, 2, 3, 4, 5]
random.shuffle(my_list)

2. 테스트 데이터 생성 함수 만들기

# 이름과 나이를 입력받아 딕셔너리 형태로 반환하는 함수
def create_person(name, age):
    return {'name': name, 'age': age}

# 이름과 성별을 입력받아 User 객체를 생성하는 함수
class User:
    def __init__(self, name, gender):
        self.name = name
        self.gender = gender

def create_user(name, gender):
    return User(name, gender)

3. 외부 데이터 사용하기

외부 데이터를 사용하여 테스트 데이터를 생성하는 방법도 있습니다. 예를 들어, CSV 파일에서 데이터를 읽어와 테스트에 활용할 수 있습니다.

import csv

def create_test_data_from_csv(csv_file):
    test_data = []
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            test_data.append(row)
    return test_data

결론

파이썬을 사용하면 다양한 방법으로 테스트 데이터를 동적으로 생성할 수 있습니다. 랜덤 데이터 생성, 함수를 통한 데이터 생성, 외부 데이터 사용 등 여러가지 방법을 활용하여 테스트 데이터를 다루는 능력을 키워보세요. 테스트 데이터의 품질은 프로그램의 신뢰성을 보장하는 데에 중요한 역할을 합니다.