added test for the partitioning