Adding test for grouped key range partitioner