Added a single test for testing probability of a function