Add benchmark to tests