Add a test helper to make test imports more DRY