improved the partition function by using NonEmptyLists
made the checkpoint tests local only