我正在尝试使用 python 框架进行处理 csv 文件的单元测试unittest。我想测试列名称匹配、列中的值匹配等情况。我知道有更方便的库,例如datatest和pytest,但我只能unittest在我的项目中使用。
猜猜我使用了错误的unittest.TestCase方法,并以错误的格式发送数据。请建议如何做得更好。
db.csv 示例:
TIMESTAMP TYPE VALUE YEAR FILE SHEET
0 02-09-2018 Index 45 2018 tq.xls A01
1 13-05-2018 Index 21 2018 tq.xls A01
2 22-01-2019 Index 9 2019 aq.xls B02
Run Code Online (Sandbox Code Playgroud)
这是代码示例:
import pandas as pd
import unittest
class DFTests(unittest.TestCase):
def setUp(self):
test_file_name = 'db.csv'
try:
data = pd.read_csv(test_file_name,
sep = ',',
header = 0)
except IOError:
print('cannot open file')
self.fixture = data
#Check column names …Run Code Online (Sandbox Code Playgroud)