add swap rebalance tests