test partition weight of 3 for partition 0
test partition weight of 3 for partition 0, with 5 partitions

So, node a should take both partition 0 (which is heavy) and partition
4 (lite), while node b takes the rest of the lite partitions 1, 2, 3.