unit test changed to take positive probabilities