style in tests; fix PartitionBy; fix iterator_to_array with Limit/Infinite iterator