chore: minor edit to unit test language marks

Author: MeanderingProgrammer

tests/callout_spec.lua

@@ -45,7 +45,7 @@ describe('callout.md', function()
             util.quote(row:increment(), '%s', ok),
             util.quote(row:increment(), '%s ', ok),
             util.code_row(row:get(), 2),
-            util.code_language(row:get(), 5, 8, 'lua'),
+            util.code_language(row:get(), 2, 'lua'),
             util.quote(row:increment(), '%s ', ok),
             util.code_row(row:get(), 2),
             util.quote(row:increment(), '%s ', ok),

tests/code_spec.lua

@@ -49,7 +49,7 @@ describe('code.md', function()
 
         vim.list_extend(expected, {
             util.code_row(row:increment(2), 0),
-            util.code_language(row:get(), 3, 7, 'rust'),
+            util.code_language(row:get(), 0, 'rust'),
         })
         for _ = 1, 3 do
             table.insert(expected, util.code_row(row:increment(), 0))
@@ -59,7 +59,7 @@ describe('code.md', function()
         vim.list_extend(expected, {
             util.bullet(row:increment(2), 0, 1),
             util.code_row(row:increment(2), 2),
-            util.code_language(row:get(), 5, 8, 'lua'),
+            util.code_language(row:get(), 2, 'lua'),
         })
         for _ = 1, 2 do
             table.insert(expected, util.code_row(row:increment(), 2))
@@ -69,7 +69,7 @@ describe('code.md', function()
         vim.list_extend(expected, {
             util.bullet(row:increment(2), 0, 1),
             util.code_row(row:increment(2), 2),
-            util.code_language(row:get(), 5, 8, 'lua'),
+            util.code_language(row:get(), 2, 'lua'),
         })
         for _, col in ipairs({ 2, 0, 2 }) do
             table.insert(expected, util.code_row(row:increment(), col))
@@ -95,7 +95,7 @@ describe('code.md', function()
         vim.list_extend(expected, {
             util.code_row(row:increment(2), 0),
             hide_background(row:get(), 0, 34),
-            util.code_language(row:get(), 3, 7, 'rust'),
+            util.code_language(row:get(), 0, 'rust'),
         })
         for _ = 1, 3 do
             vim.list_extend(expected, {
@@ -110,7 +110,7 @@ describe('code.md', function()
             util.bullet(row:increment(2), 0, 1),
             util.code_row(row:increment(2), 2),
             hide_background(row:get(), 2, 20),
-            util.code_language(row:get(), 5, 8, 'lua'),
+            util.code_language(row:get(), 2, 'lua'),
         })
         for _ = 1, 2 do
             vim.list_extend(expected, {
@@ -125,7 +125,7 @@ describe('code.md', function()
             util.bullet(row:increment(2), 0, 1),
             util.code_row(row:increment(2), 2),
             hide_background(row:get(), 2, 20),
-            util.code_language(row:get(), 5, 8, 'lua'),
+            util.code_language(row:get(), 2, 'lua'),
         })
         for _, col in ipairs({ 2, 0, 2 }) do
             vim.list_extend(expected, {

tests/heading_code_spec.lua

@@ -20,7 +20,7 @@ describe('heading_code.md', function()
 
         vim.list_extend(expected, {
             util.code_row(row:increment(2), 0),
-            util.code_language(row:get(), 3, 9, 'python'),
+            util.code_language(row:get(), 0, 'python'),
         })
         for _ = 13, 21 do
             table.insert(expected, util.code_row(row:increment(), 0))

tests/util.lua

@@ -137,11 +137,10 @@ function M.code_row(row, col)
 end
 
 ---@param row integer
----@param start_col integer
----@param end_col integer
+---@param col integer
 ---@param name 'python'|'lua'|'rust'
 ---@return render.md.MarkInfo[]
-function M.code_language(row, start_col, end_col, name)
+function M.code_language(row, col, name)
     local icon, highlight
     if name == 'python' then
         icon, highlight = '󰌠 ', 'MiniIconsYellow'
@@ -153,14 +152,14 @@ function M.code_language(row, start_col, end_col, name)
     ---@type render.md.MarkInfo
     local sign_mark = {
         row = { row, row },
-        col = { start_col, end_col },
+        col = { col + 3, col + 3 + #name },
         sign_text = icon,
         sign_hl_group = M.hl('_' .. highlight .. '_' .. M.hl('Sign')),
     }
     ---@type render.md.MarkInfo
     local language_mark = {
         row = { row },
-        col = { start_col },
+        col = { col + 3 },
         virt_text = { { icon .. name, { highlight, M.hl('Code') } } },
         virt_text_pos = 'inline',
     }