units: Colorize test results in new units script

Don't know if the code formatting/structure is as you like, but there general idea is in the patch.
Merge it or re-implement it as you like :)