模拟本地python变量
模拟本地python变量
提问于 2018-09-14 08:35:27
假设我有以下课程：
class Person:
def __init__(self, name):
self.name = name
def print_name(self):
date = datetime.now().strftime('%Y-%m-%d')
return 'Hello {}. Today is {}'.format(self.name, date)我试图模拟date变量，以便为print_name方法编写一个成功的测试。我一直在更改函数，以便从一个函数返回date，这很容易模拟。
class Person:
def __init__(self, name):
self.name = name
def print_name(self):
date = self._get_date_now()
return 'Hello {}. Today is {}'.format(self.name, date)
@staticmethod
def _get_date_now():
return datetime.now().strftime('%Y-%m-%d')单元测试看起来是这样的：
from mock import patch
class MyTests(unittest.TestCase):
@patch.object(Person, '_get_date_now')
def test_hello(self, date_mock):
date_mock.return_value = '2018-10-10'
person = Person('Jim')
result = person.print_name()
self.assertEqual(result, 'Hello Jim. Today is 2018-10-10')我是否可以直接模拟date变量，或者这是最好的方法。
注意，这是一个简单的例子，创建_get_date_now函数非常简单。然而，我的用例要复杂得多，模拟出这些局部变量将是非常有用的。
回答 1
Stack Overflow用户
回答已采纳
发布于 2018-09-15 01:43:25
使用冷冻枪
from freezegun import freeze_time
@freeze_time("2018-10-10")
def test_hello(self):
person = Person('Jim')
result = person.print_name()
self.assertEqual(result, 'Hello Jim. Today is 2018-10-10')页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/52328011
复制相关文章
相似问题
腾讯云开发者
