added doctests to tests