Add test cases for zmt and zmt ui modules