neovim/test/benchmark/keycodes_spec.lua

local n = require('test.functional.testnvim')()
local t = require('test.testutil')
local clear = n.clear
local api = n.api
local fn = n.fn

local keycodes = require('src.nvim.keycodes')

describe('nvim_replace_termcodes performance', function()
  it('200 calls with a key repeated 5000 times', function()
    clear()
    local stats = {}
    local ms = 1 / 1000000

    for _, keycode in ipairs(keycodes.names) do
      local notation = ('<%s>'):format(keycode[2])
      local str = notation:rep(5000)

      local start = vim.uv.hrtime()
      for _ = 1, 200 do
        api.nvim_replace_termcodes(str, false, true, true)
      end
      local elapsed = vim.uv.hrtime() - start

      table.insert(stats, elapsed)
      io.stdout:write(('\n%-20s%14.6f ms'):format(notation, elapsed * ms))
      io.stdout:flush()
    end
    io.stdout:write('\n')

    t.bench_report(stats, { unit = 'ms' })
  end)
end)

describe('keytrans() performance', function()
  it('200 calls with a key repeated 5000 times', function()
    clear()
    local stats = {}
    local ms = 1 / 1000000

    for _, keycode in ipairs(keycodes.names) do
      local notation = ('<%s>'):format(keycode[2])
      local str = api.nvim_replace_termcodes(notation, false, true, true):rep(5000)

      local start = vim.uv.hrtime()
      for _ = 1, 200 do
        fn.keytrans(str)
      end
      local elapsed = vim.uv.hrtime() - start

      table.insert(stats, elapsed)
      io.stdout:write(('\n%-20s%14.6f ms'):format(notation, elapsed * ms))
      io.stdout:flush()
    end
    io.stdout:write('\n')

    t.bench_report(stats, { unit = 'ms' })
  end)
end)