Merge "Add a unit test to check partition info"