小编干猕猴*_*猕猴桃的帖子

使用unittest框架测试pandas数据框

我正在尝试使用 python 框架进行处理 csv 文件的单元测试unittest。我想测试列名称匹配、列中的值匹配等情况。我知道有更方便的库,例如datatestpytest,但我只能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)

python unit-testing dataframe pandas python-unittest

2
推荐指数
1
解决办法
3337
查看次数