Продолжаем тему моков - side_effect для динамического поведения и autospec для защиты от самой коварной проблемы моков: теста, который проходит, хотя реальный код давно сломан.
side_effect
autospec