Extended the candidate unittests

Added 2 unittests which check probabilities of returned candidates