Enable test for stratified sampling for multi-class after load-balancing.