Add test generator
Refactor test generator
v0.1.0