Add a test for a correct lambda probability input.
Closes #11.