我想模拟特定函数中的一些嵌套函数：
# tools.py
def cpu_count():
def get_cpu_quota():
return int(load("/sys/fs/cgroup/cpu/cpu.cfs_quota_us"))
def get_cpu_period():
return int(load("/sys/fs/cgroup/cpu/cpu.cfs_period_us"))
def get_cpus():
try:
cfs_quota_us = get_cpu_quota()
cfs_period_us = get_cpu_period()
if cfs_quota_us > 0 and cfs_period_us > 0:
return int(math.ceil(cfs_quota_us / cfs_period_us))
except:
pass
return multiprocessing.cpu_count()
return get_cpus()
# test_tools.py
class TestTools(unittest.TestCase):
@patch("tools.cpu_count")
def test_cpu_count_in_container(self, cpu_count_mock):
cpu_count_mock.return_value.get_cpu_quota.return_value = 5000
cpu_count_mock.return_value.get_cpu_period.return_value = 1000
cpus = tools.cpu_count()
self.assertEqual(5, cpus)
但是，当运行此测试时，我遇到以下错误：
FAIL: test_cpu_count_in_container (tests.util.tools_test.ToolsTest)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/home/uilian/Development/foo/venv/lib/python3.7/site-packages/mock/mock.py", line 1305, in patched
return func(*args, **keywargs)
File "/home/uilian/Development/foo/tests/util/tools_test.py", line 154, in test_cpu_count_in_container
self.assertIsInstance(cpus, int)
AssertionError: <MagicMock name='cpu_count()' id='140490818322384'> is not an instance of <class 'int'>
是否可以模拟嵌套函数？
如果不是，对于这种情况最好的方法是什么？
None
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
