[M] Reformat test code