Add testcases for get partitions