python快速批量生成测试数据

用数据驱动测试时，经常需要生成批量的随机数据，可以通过Faker快速生成。

下面是示例，更多类型的测试数据可以参考官网Locale zh_CN — Faker 13.12.1 documentation

from faker import Faker

fake = Faker(locale='zh_CN')

# 姓名
for _ in range(5):
    print(fake.name())

# 身份证号
for _ in range(5):
    print(fake.ssn())

# 城市名称
for _ in range(5):
    print(fake.city_name())

# 地址
for _ in range(5):
    print(fake.address())

运行结果：

周桂荣
陈峰
王秀珍
杨勇
吴峰
110108200005026057
411525196211158684
130105194710187987
360734193409052824
211103193511112513
邯郸
武汉
海门
南京
重庆
新疆维吾尔自治区敏县白云西安街Q座 334445
辽宁省燕市翔安周路q座 100249
宁夏回族自治区瑞县淄川陈街V座 643055
黑龙江省澳门市崇文高路R座 610999
安徽省凤英市高坪重庆街b座 709958

import csv
import datetime
import os


# 用当前日期命名文件
filename = str(datetime.datetime.now()) + '.csv'

# 存储文件路径，根据自己的需要修改本地路径
file_path = os.path.join('/Users/yehl/Documents/', filename)

file = open(file_path, 'w', encoding='utf-8', newline='')

write_data = csv.writer(file)

# 随机生成的5个名字导出csv
for i in range(5):
    write_data.writerow([fake.name()])
file.close()

运行结果