Move TMS unit tests