Kann ich 'zufällig' mit unittest.mock.patch patchen?

Question

Ich bin daran interessiert, etwas Code zu testen, der das "zufällige" Modul verwendet, und ich würde gerne in der Lage sein, meine eigene falsche Version von random zu patchen/einzufügen, wenn meine Tests laufen, die einen bekannten Wert zurückgibt es zurück zum normalen zufälligen Modul danach. Aus der Dokumentation kann ich nur sehen, dass ich Klassen kitten kann. Gibt es eine Möglichkeit, Funktionen zu patchen? So etwas wie das:

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 

Dieser Code funktioniert nicht, was brauche ich stattdessen?

Answer 1

patch.function scheint nicht zu existieren. Sie können patch selbst statt:

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