def test_timedelta_type(now): expected = datetime.datetime(2014, 1, 15, 10, 10, tzinfo=tz.tzutc()) with mock.patch( 'docker_custodian.args.datetime.datetime', autospec=True, ) as mock_datetime: mock_datetime.now.return_value = now assert args.timedelta_type('5 days') == expected
def test_timedelta_type_none(): assert args.timedelta_type(None) is None