[开发测试]AllPairs的测试用例自动生成代码

import xlwt
from allpairspy import AllPairs
import re


def fread(file_path):
    # 读取文档写入list
    f = open(file_path, "r", encoding='utf-8')  # 设置文件对象
    data = f.readlines()  # 直接将文件中按行读到list里
    f.close()  # 关闭文件
    return data


def parse(data):
    # 解析/过滤读取的文件成为可识别格式
    col = []
    parameters = []
    for line in data:
        a = line.strip('\n')  # 去除每行尾的\n
        b = a.replace(' ', '')  # 去除每行中的所有空格
        col1 = re.split('[:：]', b)[0]  # 获取每行的名称
        col.append(col1)  # 获取用例的表头，数组

        c = b[b.rfind(':', 1) + 1:]  # 去除:前的所有字符
        d = c[c.rfind('：', 1) + 1:]  # 去除：前的所有字符
        e = re.split('[,，]', d)  # 根据,或，符号拆分字符
        parameters.append(e)  # 获取测试用例的内容，数组
    return parameters, col


# 打印在控制台
def allpairs1(parameters):
    p = []
    print("PAIRWISE:")
    for i, pairs in enumerate(AllPairs(parameters)):
        if pairs[1] == '男' and pairs[0] == '张三':  # 测试用例的约束条件
            continue
        elif pairs[1] == '女' and pairs[0] == '张三':  # 测试用例的约束条件
            continue

        p.append(pairs)

    return p  # 返回测试用例集数组


# 输出在新的文件中
def allpairs(name, col, datalist):
    book = xlwt.Workbook(encoding='utf-8', style_compression=0)
    sheet = book.add_sheet(name, cell_overwrite_ok=True)
    for i in range(0, len(col)):
        sheet.write(0, i, col[i])

    for i in range(0, len(datalist)):
        data = datalist[i]
        for j in range(0, len(col)):
            sheet.write(i + 1, j, data[j])
    savepath = r'{}.xls'.format(name)
    book.save(savepath)


if __name__ == "__main__":
    file_path = r'data1.txt'
    parameters = parse(fread(file_path))[0]
    col = parse(fread(file_path))[1]
    datalist = allpairs1(parameters)
    allpairs('测试用例', col, datalist)