Starting to write a test for preserving partition assignments