Add test case for probability().