Properly compute change locations for use-walrus-if #337

See the test case introduced in #336. We need to properly compute the change locations (specifically the line numbers) when lines are added by the codemod.

It's currently unclear to me whether this affects multiple codemods.