Restructured TS map computation

As close to final as it could get today.
Please extend the assert to not only cover position, but also TS value
(and if possible amplitude and n_iter)