如何在假设中传递不同的数据类型值,以及如何在Python中检查返回类型和值是否正确

2024-10-03 21:27:47 发布

您现在位置:Python中文网/ 问答频道 /正文

使用假设模块在Python中进行单元测试的示例代码。 编写了简单的getTimeDelta函数来获取两个日期之间的时间差。你知道吗

想写getTimeDelta函数的单元测试。使用假设模块获得多个datetime值。你知道吗

问题是: 1如何传递不同的数据类型值,即time1是str 而time2datetime没有写多个函数吗?你知道吗

  1. 只检查返回值的数据类型,不检查实际结果是否正确。怎么做?

  2. 如何检查项目的数据类型,结果元组的数据类型?assertIsInstance

代码:

import unittest
from hypothesis import given
from hypothesis import strategies as st
import datetime
from dateutil.parser import parse as time_parse

def getTimeDelta(time1, time2):
    try:
        if isinstance(time1, str):
            time1 = time_parse(time1)
        if isinstance(time2, str):
            time2 = time_parse(time2)
        return (time1 - time2, {"message": ""})
    except Exception as err:
        return (False, {"message": "Exception {}".format(err)})

class TestTimeDeltaCalc(unittest.TestCase):
    def __init__(self, *args, **kwargs):
        super(TestTimeDeltaCalc, self).__init__(*args, **kwargs)
        self.utils = Utils()

    @given(st.datetimes(), st.datetimes())
    def test_time_delta(self, time1, time2):
        time_delta = getTimeDelta(time1, time2)
        if time_delta[0] is False:
            self.assertIsInstance(time_delta[0], bool)
        else:
            self.assertIsInstance(time_delta[0], datetime.timedelta)

        self.assertIsInstance(time_delta[1], dict)

    @given(st.text(), st.text())
    def test_time_delta(self, time1, time2):
        time_delta = getTimeDelta(time1, time2)
        if time_delta[0] is False:
            self.assertIsInstance(time_delta[0], bool)
        else:
            self.assertIsInstance(time_delta[0], datetime.timedelta)

        self.assertIsInstance(time_delta[1], dict)


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

编辑01:

我们可以通过hypothesis.strategies.one_of(*args)link来解决问题1


Tags: 函数importselfdatetimeiftimeparsedef
1条回答
网友
1楼 · 发布于 2024-10-03 21:27:47
  1. one_of正是正确的方法。

  2. 测试属性的方法有很多(参见this article),但这里最简单的方法可能是“测试oracle”:只需将结果与已知的良好实现进行比较!否则,您可以检查timedelta是否为负/0/正,如果time1是lt/eq/gt time2-这基本上是一个较低的分辨率。

  3. 你的assertIsInstance电话好像对我有用。

相关问题 更多 >