Khan-ify partitionby test