Simplify colon-assignment-spacing linting.

And make sure it works for more test cases.