Check trajectory shape for submodules

When testing T2s submodule, shape of trajectory passed to function seems different that documentation. Documentation says param trajectories: (#reps, #particles, #steps, 3)

But actual passed shape in test case (with single time step, no reps) is (1, 1, n_part, 3)

In this case the submodules are called from _call_core of _generic.py. Are submodules called from anywhere else? Maybe shape changed when _generic was added?

In _call_core initial position is defined as (n_repetitions/1, #batch, 3) which for n_reps=1 (sequential) matches what is actually passed, no what submodule documentation suggests.