public List<Long> getAggregatedSubpartitionBytes() {
        checkState(aggregatedSubpartitionBytes != null, "Not all partition infos are ready");
        return Collections.unmodifiableList(aggregatedSubpartitionBytes);
    }
@Test
    void testGetAggregatedSubpartitionBytes() {
        AllToAllBlockingResultInfo resultInfo =
                new AllToAllBlockingResultInfo(new IntermediateDataSetID(), 2, 2, false);
        resultInfo.recordPartitionInfo(0, new ResultPartitionBytes(new long[] {32L, 64L}));
        resultInfo.recordPartitionInfo(1, new ResultPartitionBytes(new long[] {128L, 256L}));

        assertThat(resultInfo.getAggregatedSubpartitionBytes()).containsExactly(160L, 320L);
    }