diff options
Diffstat (limited to 'tests/query/highlights_spec.lua')
| -rw-r--r-- | tests/query/highlights_spec.lua | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/query/highlights_spec.lua b/tests/query/highlights_spec.lua index c3c7d153..ec3a3913 100644 --- a/tests/query/highlights_spec.lua +++ b/tests/query/highlights_spec.lua @@ -1,4 +1,4 @@ -require "nvim-treesitter.highlight" -- yes, this is necessary to set the hlmap +local ts_highlight = require "nvim-treesitter.highlight" -- yes, this is necessary to set the hlmap local highlighter = require "vim.treesitter.highlighter" local ts_utils = require "nvim-treesitter.ts_utils" local parsers = require "nvim-treesitter.parsers" @@ -70,9 +70,9 @@ local function check_assertions(file) assert.is.number(col) if hl and ts_utils.is_in_node_range(node, row, col) then local c = query._query.captures[capture] -- name of the capture in the query - if c ~= nil then + if c ~= nil and c ~= "spell" and c ~= "conceal" then captures[c] = true - local general_hl = query:_get_hl_from_capture(capture) + local general_hl = ts_highlight.default_map[c] highlights[general_hl] = true end end |