Python自带的unitTest测试数据库表数据

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import datetime
import unittest
from .PoolDB import *


class crawler_zzj_zrt_zrzjy_test(unittest.TestCase):
    def setUp(self):
        self.db = PTConnectionPool("test", "mysql.lian")#封装的数据库连接池

    def tearDown(self):
        pass

    def test_has_data(self):
        sql = "SELECT max(date) date from table"
        result = self.db.select_with_name(sql)
        y = datetime.datetime.now().year
        if result[0]['date'] is not None:
            if result[0]['date'][:-1] != str(y):
                y -= 1
            nyr = str(y) + '年'
            self.assertEqual(result[0]['date'], str(nyr), "检查是否新增数据")


if __name__ == '__main__':
    unittest.main()