add test case for partition