PartitionArrangement { Map<Address, int[]> getRemotePartitionAssignment() { return remotePartitionAssignment; } PartitionArrangement(Address[] partitionOwners, Address thisAddress); Map<Address, int[]> remotePartitionAssignmentToOne(Address target);  }
@Test public void testRemotePartitionAssignment() { assertEquals(2, a.getRemotePartitionAssignment().size()); assertArrayEquals(new int[]{2, 4}, a.getRemotePartitionAssignment().get(a1)); assertArrayEquals(new int[]{1, 5}, a.getRemotePartitionAssignment().get(a2)); }