Split out `colorization` for testing.

See #246.