remove doctest and add unittest