Update benchmarking tests to use the pedantic API

[seddonym]

Codspeed now* supports the pytest-benchmark's pedantic API, making it possibly to do setup/teardown.

We should update the benchmarking tests to use this where appropriate.

*May not be released yet.