Fix runtests for unittests
Fix colorization unittest
Address code review
Address code review
Merge pull request #924 from Microsoft/fixColorizationUnittest

Fix colorization unittest