Add pytest-benchmark to the test suite

[harshil21]

We should consider adding pytest-benchmark to the test suite. This will allow us to keep track of the performance of the code over time.

Additionally there should be another Github action to compare the latest benchmark run on the head of a branch and compare it against main. We can export the .png as artifacts to allow for easy viewing in the browser itself.