Reorganize tests, use "doctest"