ENH: Extend tests for stop_probability