单元测试python数据类型测试失败“预期：NaT Actual:NaT"

Question

我将介绍使用unittest模块对python数据管道进行单元测试

数据类对象示例：

class IsAvailable(Object)
    employee_id: int = Property()
    start_time: str = Property()

单元测试用例：

class TestIsAvailable(TestCase):
    def setUp(self):
        self.employee = pd.read_json('employee_leave.json', orient='records')
        self.isAvailable = IsAvailable()
        self.isAvailable.id = self.employee['start_time'][0]

    def test_is_available_all_day_001(self):
        assert self.isAvailable.start_time == pd.NaT

试验结果：

self = <tests.test_nodes.TestIsAvailable testMethod=test_is_available_all_day_001>

    def test_is_available_all_day_001(self):
        """test employee is available all day on specific day of the week"""
>       assert self.isAvailable.start_time == pd.NaT
E       AssertionError: assert NaT == NaT
E        +  where NaT = <IsAvailable id=1>.start_time
E        +    where <IsAvailable id=1> = <tests.test_nodes.TestIsAvailable testMethod=test_is_available_all_day_001>.isAvailable
E        +  and   NaT = pd.NaT

如何测试数据类型

Answer 1

NaN和NaT显式地不等于它们自己

您可以通过在交互模式下运行python并键入：

import pandas as pd
pd.NaT == pd.NaT

同样地

pd.NaN == pd.NaN

您可以使用

pd.isna(pd.NaT)