Simplify partition tests

Temporary variables are no longer needed now that we're using XCTAssertEqualObjects.
Merge pull request #10 from zolrath/master

Simplify partition tests