Add rebalancing tests with colocated PRs and AEQ (colocated region)

Create tests to add colocated partitioned regions during rebalance to verify co-located region does not hang.

Create similar test with AEQ (which is a colocated region) to verify that mutating region to add AEQ does not hang during rebalance.