Refactor data-driven load tests