better tests to solve partition

looking better but its not sorting properly