import unittest
import time
from unittest import TestCase
from unittest.mock import patch
def sum(a, b):
time.sleep(100) # long running process
return a + b
class TestCalculator(TestCase):
@patch('__main__.sum', return_value=5)
def test_sum(self, sum):
self.assertEqual(sum(2, 3), 5)
if __name__ == '__main__':
unittest.main()
本系列文章和代码已经作为项目归档到github，仓库地址：jumper2014/PyCodeComplete。大家觉得有帮助就请在github上star一下，你的支持是我更新的动力。什么？你没有github账号？学习Python怎么可以没有github账号呢，快去注册一个啦！