Write tests for count_loader