adding more tests to check the find best partition algorithm and making the benchmark class better testable