如何从单元测试中模拟.patch库函数

Question

我有一个名为learning的模块，它使用random.uniform()。我有一个名为test_learning.py的文件，其中包含单元测试。当我运行单元测试时，我希望learning中的代码看到random.uniform()的修补版本。我该怎么做？这是我目前拥有的。

import random
import unittest
import unittest.mock as mock
class TestLearning(unittest.TestCase):
def test_get_random_belief_bit(self):
with mock.patch('learning.random.uniform', mock_uniform):
bit = learning.get_random_belief_bit(0.4)
self.assertEqual(bit, 0)

但是测试(有时(失败learning.get_random_belief_bit()因为似乎正在使用真正的random.uniform()。

Answer 1

单元测试解决方案：

learning.py：

import random

def get_random_belief_bit(f):
return random.uniform()

test_learning.py：

import random
import unittest
import unittest.mock as mock
import learning

class TestLearning(unittest.TestCase):
def test_get_random_belief_bit(self):
with mock.patch('random.uniform', mock.Mock()) as mock_uniform:
mock_uniform.return_value = 0
bit = learning.get_random_belief_bit(0.4)
self.assertEqual(bit, 0)
mock_uniform.assert_called_once()

if __name__ == '__main__':
unittest.main()

带有覆盖率报告的单元测试结果：

.
----------------------------------------------------------------------
Ran 1 test in 0.000s
OK
Name                                          Stmts   Miss  Cover   Missing
---------------------------------------------------------------------------
src/stackoverflow/57874971/learning.py            3      0   100%
src/stackoverflow/57874971/test_learning.py      13      0   100%
---------------------------------------------------------------------------
TOTAL                                            16      0   100%