Split testing into different files/functions for clarity