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%

如何从单元测试中嘲笑库函数

1 个答案: