test: refactor: introduce and use `calculate_input_weight` helper

- Select -