Put unit test in a seperate file.