Added partition load balancing test.