Fixes to unit tests

Fixes to unit tests:
- Fixed align doctest rounding problem
- `probability.doctest` now passes
- `data.doctest` now passes