Make test code better
0.0.2