diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/print-trie.lua | 96 |
1 files changed, 48 insertions, 48 deletions
diff --git a/test/print-trie.lua b/test/print-trie.lua index 2ec1322..7575b5c 100644 --- a/test/print-trie.lua +++ b/test/print-trie.lua @@ -1,62 +1,62 @@ -- TODO this is kinda shitty -local function dirname(str,sep) - sep=sep or'/' - return str:match("(.*"..sep..")") +local function dirname(str, sep) + sep = sep or "/" + return str:match("(.*" .. sep .. ")") end local script_dir = dirname(arg[0]) -package.path = script_dir.."/../lua/?.lua;"..package.path +package.path = script_dir .. "/../lua/?.lua;" .. package.path -local Trie = require 'trie' -local nvim = require 'nvim' +local Trie = require "trie" +local nvim = require "nvim" local function print_color_trie() - local tohex = bit.tohex - local min, max = math.min, math.max + local tohex = bit.tohex + local min, max = math.min, math.max - local COLOR_NAME_SETTINGS = { - lowercase = false; - strip_digits = true; - } - local COLOR_MAP = {} - local COLOR_TRIE = Trie() - for k, v in pairs(nvim.get_color_map()) do - if not (COLOR_NAME_SETTINGS.strip_digits and k:match("%d+$")) then - COLOR_NAME_MINLEN = COLOR_NAME_MINLEN and min(#k, COLOR_NAME_MINLEN) or #k - COLOR_NAME_MAXLEN = COLOR_NAME_MAXLEN and max(#k, COLOR_NAME_MAXLEN) or #k - COLOR_MAP[k] = tohex(v, 6) - COLOR_TRIE:insert(k) - if COLOR_NAME_SETTINGS.lowercase then - local lowercase = k:lower() - COLOR_MAP[lowercase] = tohex(v, 6) - COLOR_TRIE:insert(lowercase) - end - end - end - print(COLOR_TRIE) + local COLOR_NAME_SETTINGS = { + lowercase = false, + strip_digits = true, + } + local COLOR_MAP = {} + local COLOR_TRIE = Trie() + for k, v in pairs(nvim.get_color_map()) do + if not (COLOR_NAME_SETTINGS.strip_digits and k:match "%d+$") then + COLOR_NAME_MINLEN = COLOR_NAME_MINLEN and min(#k, COLOR_NAME_MINLEN) or #k + COLOR_NAME_MAXLEN = COLOR_NAME_MAXLEN and max(#k, COLOR_NAME_MAXLEN) or #k + COLOR_MAP[k] = tohex(v, 6) + COLOR_TRIE:insert(k) + if COLOR_NAME_SETTINGS.lowercase then + local lowercase = k:lower() + COLOR_MAP[lowercase] = tohex(v, 6) + COLOR_TRIE:insert(lowercase) + end + end + end + print(COLOR_TRIE) end local trie = Trie { - "cat"; - "car"; - "celtic"; - "carb"; - "carb0"; - "CART0"; - "CaRT0"; - "Cart0"; - "931"; - "191"; - "121"; - "cardio"; - "call"; - "calcium"; - "calciur"; - "carry"; - "dog"; - "catdog"; + "cat", + "car", + "celtic", + "carb", + "carb0", + "CART0", + "CaRT0", + "Cart0", + "931", + "191", + "121", + "cardio", + "call", + "calcium", + "calciur", + "carry", + "dog", + "catdog", } print(trie) -print("catdo", trie:longest_prefix("catdo")) -print("catastrophic", trie:longest_prefix("catastrophic")) +print("catdo", trie:longest_prefix "catdo") +print("catastrophic", trie:longest_prefix "catastrophic") |