Simplify partition tests

Temporary variables are no longer needed now that we're using XCTAssertEqualObjects.