Refactor: scheduler: functionize checking whether colocation applies

This allows us to have more consistent trace messages, and will reduce
code duplication when more checks are added later.