did you mean ``assert_called_with`` ?