Data.List.partition on infinite lists