wavefunction: added missing imports
ground_state: added support for N=0 in get_TF_state()
test/visibility: updated for the new interface