module RuboCop::Cop::Style::ConditionalCorrectorHelper

def replace_branch_assignment(corrector, branch)
  _variable, *_operator, assignment = *branch
  source = assignment.source
  replacement = if assignment.array_type? && !assignment.bracketed?
                  "[#{source}]"
                else
                  source
                end
  corrector.replace(branch.source_range, replacement)
end