// TODO: note that we are currently double-counting the memory usage
        // because we reserve the memory twice - once at the repartitioner level
        // and then again in each PartitionBuffer
        assert_eq!(212992, repartitioner.reservation.size());
        assert_eq!(
            106496,
            repartitioner.buffered_partitions[0].reservation.size()
        );
        assert_eq!(
            106496,
            repartitioner.buffered_partitions[1].reservation.size()
        );