Se este é o código python, como parece para mim, o meu plugin splitjoin parece fazer a coisa certa para todos desses exemplos, com algumas diferenças de recuo (que vêm do script de indentação integrado de qualquer maneira).
Isso não é tanto uma resposta em termos de "é assim que você faz", porque é um problema complicado, quando você precisa contar parênteses, ignorar strings e assim por diante. É por isso que eu escrevi um plugin inteiro para essas coisas, que lida com cada situação com uma função de callback diferente. Só posso sugerir que você experimente e veja se ele se encaixa bem com seus casos de uso.