add performance tests

[jmoralez]

Description

Adds tests to prevent introducing performance regressions.

Checklist:

• [x] This PR has a meaningful title and a clear description.
• [x] The tests pass.
• [x] All linting tasks pass.
• [x] The notebooks are clean.