Move TestBuilder to __init__.py