Don't you also need to match for cases that are not a CoPartitioner? In the case of normal group by, we can see a HashPartitioner here also (and in the future maybe other Partitioner's).