* Refactor UTs to cleanup UTs and increase UT coverages * User assert for consistency * Applied comments and resolve lint error * Delete unnecessary license header Co-authored-by: JungukCho <jungukcho@microsoft.com>