FIX doctest
TST improve coverage