首页 > 解决方案 > 如何在蝗虫性能测试脚本中从 csv 文件中读取数据?

问题描述

我正在尝试使用以下代码从包含 1 行和 5 列的 csv 文件中读取数据

def __init__(self):
        super(data, self).__init__()
        global data
        if (data == None):
            with open('var.csv', 'r') as l:
                reader = csv.reader(l)
                data = list(reader)
def on_start(self):
            if len(data) > 0:
                self.my_value = data.pop()

我的输出是 ('sample') 我希望它是样本

标签: locust

解决方案


将最后一行从 更改self.my_value = data.pop()self.my_value = data.pop()[0]

但您也可以使用 locust 插件 csv 阅读器:https ://github.com/SvenskaSpel/locust-plugins/blob/master/examples/csvreader_ex.py


推荐阅读