Test class separation logic which can be customized