change partitioning strategy to declare type that it will return
update tests to reflect new partition method