读取csv文件的函数
def getCsvData(file): # 读取CSV文件 value_rows = [] base_dir = str(os.path.dirname(os.path.dirname(__file__))) base_dir = base_dir.replace('\\', '/') csv_path = base_dir + file # csv路径 with open(csv_path) as f: # , encoding='UTF-8' f_csv = csv.reader(f) next(f_csv) for r in f_csv: value_rows.append(r) print(value_rows) return value_rows
@ddt class RyomsTest(unittest.TestCase): """大运营模块测试""" def setUp(self): pass def tearDown(self): pass @data(*getCsvData("/data/ryoms_app_center.csv")) @unpack # 解包 # @file_data("/data/ryoms_app_center.csv") def test_easy_smart_basic_module_mall(self, data, alert_data): """应用中心-参数配置-商场配置-获取商场配置接口/easy-smart-basic/module/mall""" print(data) url = env_url + "/easy-smart-basic/module/mall" data = eval(data) # 去除csv带来的""引号 r = requests.post(url, headers=header, json=data) print(r.status_code) print(r.json()) result = r.json() self.assertEqual(r.status_code, 200) self.assertEqual(result['meta']['msg'], alert_data) self.assertEqual(result['success'], True)