Added TestRebalance test case and more test inputs