unittest.mock.patchを使って 'random'にパッチを当てることはできますか？

Question

私は 'ランダム'モジュールを使用するいくつかのコードをテストすることに興味があり、テストが実行されているときに私自身の偽のバージョンをランダムに挿入することができます後で通常のランダムモジュールに戻ります。ドキュメントから、私はクラスにパッチを当てることができます。関数にパッチを当てる方法はありますか？このようなもの：

def my_code_that_uses_random(): 
    return random.choice([0, 1, 2, 3]) 

with patch.function(random.choice, return_value=3) as mock_random: 
    choice = my_code_that_uses_random() 
    assert choice == 3 

そのコードは機能しませんが、代わりに何が必要ですか？

Answer 1

patch.functionは存在しないようです。代わりにpatch自体を使用することができます。

with patch('random.choice', return_value=3) as mock_random: 
    choice = my_code_that_uses_random() 
    assert choice == 3