Jupyter is not linear. You can edit and evaluate any cell at any time. If you don't want test interspersed with the code, you can keep tests at the bottom by adding cells in the middle.