Refactored partition impl.
Added check for required unsafe methods. Skipped a partition state generator test.