Frozen_time无法用于默认参数

时间:2018-09-26 06:37:28

标签: python-unittest freezegun

我有一个函数,其参数的默认值为datetime.now()。方法如下所示

def as_standard_format(p_date=datetime.now(), fmt=sdk_constants.DEFAULT_DATE_TIME_FORMAT):
    return p_date.strftime(fmt)

我有一种类似于下面的测试方法,

@freeze_time(datetime(year=2018, month=9, day=25, hour=15, minute=46, second=36))
def testNowWithDefaultFormat(self):

    # The below method gives the frozen time
    print(datetime.now())

    # But the below test fails
    assert utils.format_date(
        datetime.now()) == "20180925T154636Z", "Default call not giving current time in " \
                                               "standard format \"%Y%m%dT%H%M%SZ\""

为什么freeze_time不能使用默认参数值?

1 个答案:

答案 0 :(得分:1)

您是Python中common gotcha的受害者,该变量与可变的默认值参数有关,而不仅限于freeze_time

有关说明,请考虑以下代码:

import time from datetime import datetime

def what_time_is_it(p_date=datetime.now()):
    return p_date.strftime('"%Y-%m-%d %X"')

if __name__ == '__main__':
    print(what_time_is_it())
    time.sleep(10)
    print(what_time_is_it())

您可能期望发生的事情

"2019-02-28 13:47:24"
"2019-02-28 13:47:34" # This line showing a datetime 10 seconds after the first one

每次调用datetime.now()都会调用一个新的what_time_is_it()

会发生什么

"2019-02-28 13:47:24"
"2019-02-28 13:47:24" # EXACTLY the same time in both lines!

定义函数后,将一次创建新的datetime.now(),并且在每个后续调用中使用相同的值。

您应该怎么做?

每次调用该函数时,通过使用默认arg信号表示未提供任何参数来创建一个新对象(通常是一个不错的选择):

def as_standard_format(p_date=None, fmt=sdk_constants.DEFAULT_DATE_TIME_FORMAT):
    if p_date is None:
        return datetime.now().strftime(fmt)
    else:
        return p_date.strftime(fmt)