Custom use statements can now be set into generated units tests