diff --git a/lua/copilot/api/init.lua b/lua/copilot/api/init.lua
index 9a556396..898fd984 100644
--- a/lua/copilot/api/init.lua
+++ b/lua/copilot/api/init.lua
@@ -130,7 +130,7 @@ function M.notify_shown(client, params, callback)
   return M.request(client, "notifyShown", params, callback)
 end
 
----@alias copilot_get_completions_data_completion { displayText: string, position: { character: integer, line: integer }, range: { ['end']: { character: integer, line: integer }, start: { character: integer, line: integer } }, text: string, uuid: string }
+---@alias copilot_get_completions_data_completion { displayText: string, position: { character: integer, line: integer }, range: { ['end']: { character: integer, line: integer }, start: { character: integer, line: integer } }, text: string, uuid: string, partial_text: string }
 ---@alias copilot_get_completions_data { completions: copilot_get_completions_data_completion[] }
 
 ---@return any|nil err
diff --git a/lua/copilot/suggestion/init.lua b/lua/copilot/suggestion/init.lua
index 1cbfebed..0fa5d309 100644
--- a/lua/copilot/suggestion/init.lua
+++ b/lua/copilot/suggestion/init.lua
@@ -9,7 +9,7 @@ local utils = require("copilot.client.utils")
 
 local M = {}
 
----@alias copilot_suggestion_context { first?: integer, cycling?: integer, cycling_callbacks?: (fun(ctx: copilot_suggestion_context):nil)[], params?: table, suggestions?: copilot_get_completions_data_completion[], choice?: integer, shown_choices?: table<string, true> }
+---@alias copilot_suggestion_context { first?: integer, cycling?: integer, cycling_callbacks?: (fun(ctx: copilot_suggestion_context):nil)[], params?: table, suggestions?: copilot_get_completions_data_completion[], choice?: integer, shown_choices?: table<string, true>, accepted_partial?: boolean }
 
 local copilot = {
   setup_done = false,
@@ -26,6 +26,8 @@ local copilot = {
   debounce = 75,
 }
 
+local ignore_next_cursor_moved = false
+
 local function with_client(fn)
   local client = c.get()
   if client then
@@ -58,6 +60,28 @@ local function get_ctx(bufnr)
   return ctx
 end
 
+---@param idx integer
+---@param new_line integer
+---@param new_end_col integer
+---@param bufnr? integer
+local function update_ctx_suggestion_position(idx, new_line, new_end_col, bufnr)
+  bufnr = bufnr or vim.api.nvim_get_current_buf()
+
+  if not copilot.context[bufnr] then
+    return
+  end
+
+  if not copilot.context[bufnr].suggestions[idx] then
+    return
+  end
+
+  local suggestion = copilot.context[bufnr].suggestions[idx]
+  suggestion.range["start"].line = new_line
+  suggestion.range["start"].character = 0
+  suggestion.range["end"].line = new_line
+  suggestion.range["end"].character = new_end_col
+end
+
 ---@param idx integer
 ---@param text string
 ---@param bufnr? integer
@@ -89,6 +113,7 @@ local function reset_ctx(ctx)
   ctx.suggestions = nil
   ctx.choice = nil
   ctx.shown_choices = nil
+  ctx.accepted_partial = nil
 end
 
 local function set_keymap(keymap)
@@ -249,10 +274,10 @@ local function get_current_suggestion(ctx)
       return nil
     end
 
-    if choice.range.start.character ~= 0 then
-      -- unexpected range
-      return nil
-    end
+    -- if choice.range.start.character ~= 0 then
+    --   -- unexpected range
+    --   return nil
+    -- end
 
     return choice
   end)
@@ -475,7 +500,7 @@ local function schedule(ctx)
     stop_timer()
   end
 
-  update_preview(ctx)
+  -- update_preview(ctx)
   local bufnr = vim.api.nvim_get_current_buf()
   copilot._copilot_timer = vim.fn.timer_start(copilot.debounce, function(timer)
     logger.trace("suggestion schedule timer", bufnr)
@@ -487,6 +512,10 @@ function M.next()
   local ctx = get_ctx()
   logger.trace("suggestion next", ctx)
 
+  if ctx.accepted_partial then
+    reset_ctx(ctx)
+  end
+
   -- no suggestion request yet
   if not ctx.first then
     logger.trace("suggestion next, no first request")
@@ -503,6 +532,10 @@ function M.prev()
   local ctx = get_ctx()
   logger.trace("suggestion prev", ctx)
 
+  if ctx.accepted_partial then
+    reset_ctx(ctx)
+  end
+
   -- no suggestion request yet
   if not ctx.first then
     logger.trace("suggestion prev, no first request", ctx)
@@ -532,13 +565,17 @@ function M.accept(modifier)
     return
   end
 
-  cancel_inflight_requests(ctx)
-  reset_ctx(ctx)
-
   if type(modifier) == "function" then
     suggestion = modifier(suggestion)
   end
 
+  local accepted_partial = suggestion.partial_text and suggestion.partial_text ~= ""
+
+  if not accepted_partial then
+    cancel_inflight_requests(ctx)
+    reset_ctx(ctx)
+  end
+
   with_client(function(client)
     local ok, _ = pcall(function()
       api.notify_accepted(
@@ -552,34 +589,62 @@ function M.accept(modifier)
     end
   end)
 
-  clear_preview()
+  local newText
+
+  if accepted_partial then
+    newText = suggestion.partial_text
+    ctx.accepted_partial = true
+    ignore_next_cursor_moved = true
+  else
+    clear_preview()
+    newText = suggestion.text
+  end
 
-  local range, newText = suggestion.range, suggestion.text
+  local range = suggestion.range
   local cursor = vim.api.nvim_win_get_cursor(0)
   local line, character = cursor[1] - 1, cursor[2]
   if range["end"].line == line and range["end"].character < character then
     range["end"].character = character
   end
 
-  -- Hack for 'autoindent', makes the indent persist. Check `:help 'autoindent'`.
   vim.schedule_wrap(function()
     -- Create an undo breakpoint
     vim.cmd("let &undolevels=&undolevels")
+    -- Hack for 'autoindent', makes the indent persist. Check `:help 'autoindent'`.
     vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes("<Space><Left><Del>", true, false, true), "n", false)
     local bufnr = vim.api.nvim_get_current_buf()
+
     local encoding = vim.api.nvim_get_option_value("fileencoding", { buf = bufnr }) ~= ""
         and vim.api.nvim_get_option_value("fileencoding", { buf = bufnr })
       or vim.api.nvim_get_option_value("encoding", { scope = "global" })
-    vim.lsp.util.apply_text_edits({ { range = range, newText = newText } }, bufnr, encoding)
 
-    -- instead of calling <End>, go to the pos of the row after the last \n of inserted text
-    -- local cursor_keys = string.rep("<Down>", #vim.split(newText, "\n", { plain = true }) - 1) .. "<End>"
-    -- vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes(cursor_keys, true, false, true), "n", false)
     local lines = vim.split(newText, "\n", { plain = true })
-    local last_line = lines[#lines]
-    local cursor_keys = string.rep("<Down>", #lines - 1)
+    local lines_count = #lines
+    local last_col = #lines[lines_count]
+
+    -- apply_text_edits will remove the last \n if the last line is empty,
+    -- so we trick it by adding an extra one
+    if last_col == 0 then
+      newText = newText .. "\n"
+    end
+
+    vim.lsp.util.apply_text_edits({ { range = range, newText = newText } }, bufnr, encoding)
+
     -- Position cursor at the end of the last inserted line
-    vim.api.nvim_win_set_cursor(0, { range["start"].line + #lines, #last_line })
+    local new_cursor_line = range["start"].line + #lines
+    vim.api.nvim_win_set_cursor(0, { new_cursor_line, last_col })
+
+    if accepted_partial then
+      suggestion.partial_text = nil
+
+      for _ = 1, lines_count - 1 do
+        suggestion.text = suggestion.text:sub(suggestion.text:find("\n") + 1)
+        suggestion.displayText = suggestion.displayText:sub(suggestion.displayText:find("\n") + 1)
+      end
+
+      update_ctx_suggestion_position(ctx.choice, new_cursor_line - 1, last_col, bufnr)
+      update_preview(ctx)
+    end
   end)()
 end
 
@@ -592,19 +657,18 @@ function M.accept_word()
 
     local _, char_idx = string.find(text, "%s*%p*[^%s%p]*%s*", character + 1)
     if char_idx then
-      suggestion.text = string.sub(text, 1, char_idx)
-
-      range["end"].line = range["start"].line
+      suggestion.partial_text = string.sub(text, 1, char_idx)
       range["end"].character = char_idx
     end
 
+    range["end"].line = range["start"].line
     return suggestion
   end)
 end
 
 function M.accept_line()
   M.accept(function(suggestion)
-    local text = suggestion.text
+    local range, text = suggestion.range, suggestion.text
 
     local cursor = vim.api.nvim_win_get_cursor(0)
     local _, character = cursor[1], cursor[2]
@@ -612,9 +676,11 @@ function M.accept_line()
     local next_char = string.sub(text, character + 1, character + 1)
     local _, char_idx = string.find(text, next_char == "\n" and "\n%s*[^\n]*\n%s*" or "\n%s*", character)
     if char_idx then
-      suggestion.text = string.sub(text, 1, char_idx)
+      suggestion.partial_text = string.sub(text, 1, char_idx)
+      range["end"].character = char_idx
     end
 
+    range["end"].line = range["start"].line
     return suggestion
   end)
 end
@@ -659,6 +725,11 @@ local function on_buf_enter()
 end
 
 local function on_cursor_moved_i()
+  if ignore_next_cursor_moved then
+    ignore_next_cursor_moved = false
+    return
+  end
+
   local ctx = get_ctx()
   if copilot._copilot_timer or ctx.params or should_auto_trigger() then
     logger.trace("suggestion on cursor moved insert")
diff --git a/tests/child_helper.lua b/tests/child_helper.lua
index 8e57eb95..670de8e4 100644
--- a/tests/child_helper.lua
+++ b/tests/child_helper.lua
@@ -21,7 +21,7 @@ function M.new_child_neovim(test_name)
       panel = "",
       suggestion = [[
         suggestion_notification = function(virt_text, _)
-          if (#virt_text > 0) and (#virt_text[1] > 0) and (virt_text[1][1] == "89") then
+          if (#virt_text > 0) and (#virt_text[1] > 0) then
             M.suggested = true
           end
         end,
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-then-accept b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-then-accept
new file mode 100644
index 00000000..7250cfed
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-then-accept
@@ -0,0 +1,103 @@
+--|---------|---------|---------|---------|
+01|# Numbers in a 3x3 grid, up to 63       
+02|{                                       
+03|  1,2,3                                 
+04|  4,5,6                                 
+05|  7,8,9                                 
+06|}                                       
+07|{                                       
+08|  10,11,12                              
+09|  13,14,15                              
+10|  16,17,18                              
+11|}                                       
+12|{                                       
+13|  19,20,21                              
+14|  22,23,24                              
+15|  25,26,27                              
+16|}                                       
+17|{                                       
+18|  28,29,30                              
+19|  31,32,33                              
+20|  34,35,36                              
+21|}                                       
+22|{                                       
+23|  37,38,39                              
+24|  40,41,42                              
+25|  43,44,45                              
+26|}                                       
+27|{                                       
+28|  46,47,48                              
+29|  49,50,51                              
+30|  52,53,54                              
+31|}                                       
+32|{                                       
+33|  55,56,57                              
+34|  58,59,60                              
+35|  61,62,63                              
+36|}                                       
+37|~                                       
+38|~                                       
+39|~                                       
+40|~                                       
+41|~                                       
+42|~                                       
+43|~                                       
+44|~                                       
+45|~                                       
+46|~                                       
+47|~                                       
+48|~                                       
+49|[No Name] [+]         36,2           All
+50|-- INSERT --                            
+
+--|---------|---------|---------|---------|
+01|0000000000000000000000000000000000000000
+02|0000000000000000000000000000000000000000
+03|0000000000000000000000000000000000000000
+04|0000000000000000000000000000000000000000
+05|0000000000000000000000000000000000000000
+06|0000000000000000000000000000000000000000
+07|0000000000000000000000000000000000000000
+08|0000000000000000000000000000000000000000
+09|0000000000000000000000000000000000000000
+10|0000000000000000000000000000000000000000
+11|0000000000000000000000000000000000000000
+12|0000000000000000000000000000000000000000
+13|0000000000000000000000000000000000000000
+14|0000000000000000000000000000000000000000
+15|0000000000000000000000000000000000000000
+16|0000000000000000000000000000000000000000
+17|0000000000000000000000000000000000000000
+18|0000000000000000000000000000000000000000
+19|0000000000000000000000000000000000000000
+20|0000000000000000000000000000000000000000
+21|0000000000000000000000000000000000000000
+22|0000000000000000000000000000000000000000
+23|0000000000000000000000000000000000000000
+24|0000000000000000000000000000000000000000
+25|0000000000000000000000000000000000000000
+26|0000000000000000000000000000000000000000
+27|0000000000000000000000000000000000000000
+28|0000000000000000000000000000000000000000
+29|0000000000000000000000000000000000000000
+30|0000000000000000000000000000000000000000
+31|0000000000000000000000000000000000000000
+32|1000000000000000000000000000000000000000
+33|0000000000000000000000000000000000000000
+34|0000000000000000000000000000000000000000
+35|0000000000000000000000000000000000000000
+36|1000000000000000000000000000000000000000
+37|2222222222222222222222222222222222222222
+38|2222222222222222222222222222222222222222
+39|2222222222222222222222222222222222222222
+40|2222222222222222222222222222222222222222
+41|2222222222222222222222222222222222222222
+42|2222222222222222222222222222222222222222
+43|2222222222222222222222222222222222222222
+44|2222222222222222222222222222222222222222
+45|2222222222222222222222222222222222222222
+46|2222222222222222222222222222222222222222
+47|2222222222222222222222222222222222222222
+48|2222222222222222222222222222222222222222
+49|3333333333333333333333333333333333333333
+50|4444444444445555555555555555555555555555
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-then-dismiss b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-then-dismiss
new file mode 100644
index 00000000..09c51635
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-then-dismiss
@@ -0,0 +1,63 @@
+--|---------|-----
+01|{              
+02|  1,2,3        
+03|  4,5,6        
+04|  7,8,9        
+05|}              
+06|{              
+07|  10,11,12     
+08|  13,14,15     
+09|  16,17,18     
+10|}              
+11|{              
+12|  19,20,21     
+13|               
+14|~              
+15|~              
+16|~              
+17|~              
+18|~              
+19|~              
+20|~              
+21|~              
+22|~              
+23|~              
+24|~              
+25|~              
+26|~              
+27|~              
+28|~              
+29|<e] [+] 13,3   
+30|-- INSERT --   
+
+--|---------|-----
+01|000000000000000
+02|000000000000000
+03|000000000000000
+04|000000000000000
+05|000000000000000
+06|000000000000000
+07|000000000000000
+08|000000000000000
+09|000000000000000
+10|000000000000000
+11|000000000000000
+12|000000000000000
+13|000000000000000
+14|111111111111111
+15|111111111111111
+16|111111111111111
+17|111111111111111
+18|111111111111111
+19|111111111111111
+20|111111111111111
+21|111111111111111
+22|111111111111111
+23|111111111111111
+24|111111111111111
+25|111111111111111
+26|111111111111111
+27|111111111111111
+28|111111111111111
+29|222222222222222
+30|333333333333444
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-works b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-works
new file mode 100644
index 00000000..ee1ebfad
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-1-line,-works
@@ -0,0 +1,63 @@
+--|---------|-----
+01|{              
+02|  1,2,3        
+03|  4,5,6        
+04|  7,8,9        
+05|}              
+06|{              
+07|  10,11,12     
+08|  13,14,15     
+09|  16,17,18     
+10|}              
+11|{              
+12|  19,20,21     
+13|               
+14|~              
+15|~              
+16|~              
+17|~              
+18|~              
+19|~              
+20|~              
+21|~              
+22|~              
+23|~              
+24|~              
+25|~              
+26|~              
+27|~              
+28|~              
+29|<e] [+] 13,2   
+30|               
+
+--|---------|-----
+01|000000000000000
+02|000000000000000
+03|000000000000000
+04|000000000000000
+05|000000000000000
+06|000000000000000
+07|000000000000000
+08|000000000000000
+09|000000000000000
+10|000000000000000
+11|000000000000000
+12|000000000000000
+13|000000000000000
+14|111111111111111
+15|111111111111111
+16|111111111111111
+17|111111111111111
+18|111111111111111
+19|111111111111111
+20|111111111111111
+21|111111111111111
+22|111111111111111
+23|111111111111111
+24|111111111111111
+25|111111111111111
+26|111111111111111
+27|111111111111111
+28|111111111111111
+29|222222222222222
+30|333333333333333
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-3-lines,-works b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-3-lines,-works
new file mode 100644
index 00000000..9a48d918
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_line,-3-lines,-works
@@ -0,0 +1,103 @@
+--|---------|-----
+01|{              
+02|  1,2,3        
+03|  4,5,6        
+04|  7,8,9        
+05|}              
+06|{              
+07|  10,11,12     
+08|  13,14,15     
+09|  16,17,18     
+10|}              
+11|{              
+12|  19,20,21     
+13|  22,23,24     
+14|  25,26,27     
+15|               
+16|~              
+17|~              
+18|~              
+19|~              
+20|~              
+21|~              
+22|~              
+23|~              
+24|~              
+25|~              
+26|~              
+27|~              
+28|~              
+29|~              
+30|~              
+31|~              
+32|~              
+33|~              
+34|~              
+35|~              
+36|~              
+37|~              
+38|~              
+39|~              
+40|~              
+41|~              
+42|~              
+43|~              
+44|~              
+45|~              
+46|~              
+47|~              
+48|~              
+49|<e] [+] 15,0-1 
+50|               
+
+--|---------|-----
+01|000000000000000
+02|000000000000000
+03|000000000000000
+04|000000000000000
+05|000000000000000
+06|000000000000000
+07|000000000000000
+08|000000000000000
+09|000000000000000
+10|000000000000000
+11|000000000000000
+12|000000000000000
+13|000000000000000
+14|000000000000000
+15|000000000000000
+16|111111111111111
+17|111111111111111
+18|111111111111111
+19|111111111111111
+20|111111111111111
+21|111111111111111
+22|111111111111111
+23|111111111111111
+24|111111111111111
+25|111111111111111
+26|111111111111111
+27|111111111111111
+28|111111111111111
+29|111111111111111
+30|111111111111111
+31|111111111111111
+32|111111111111111
+33|111111111111111
+34|111111111111111
+35|111111111111111
+36|111111111111111
+37|111111111111111
+38|111111111111111
+39|111111111111111
+40|111111111111111
+41|111111111111111
+42|111111111111111
+43|111111111111111
+44|111111111111111
+45|111111111111111
+46|111111111111111
+47|111111111111111
+48|111111111111111
+49|222222222222222
+50|333333333333333
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-then-accept b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-then-accept
new file mode 100644
index 00000000..f22c39f8
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-then-accept
@@ -0,0 +1,23 @@
+--|---------|-----
+01|1, 2, 3,       
+02|4, 5, 6,       
+03|7, 8, 9,       
+04|~              
+05|~              
+06|~              
+07|~              
+08|~              
+09|<e] [+] 3,9 All
+10|-- INSERT --   
+
+--|---------|-----
+01|000000000000000
+02|000000000000000
+03|000000000000000
+04|111111111111111
+05|111111111111111
+06|111111111111111
+07|111111111111111
+08|111111111111111
+09|222222222222222
+10|333333333333444
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-then-dismiss b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-then-dismiss
new file mode 100644
index 00000000..8351723b
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-then-dismiss
@@ -0,0 +1,23 @@
+--|---------|-----
+01|1, 2, 3,       
+02|4, 5, 6,       
+03|7, 8           
+04|~              
+05|~              
+06|~              
+07|~              
+08|~              
+09|<e] [+] 3,5 All
+10|-- INSERT --   
+
+--|---------|-----
+01|000000000000000
+02|000000000000000
+03|000000000000000
+04|111111111111111
+05|111111111111111
+06|111111111111111
+07|111111111111111
+08|111111111111111
+09|222222222222222
+10|333333333333444
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-works b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-works
new file mode 100644
index 00000000..70d3d9ac
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-1-word,-works
@@ -0,0 +1,23 @@
+--|---------|-----
+01|1, 2, 3,       
+02|4, 5, 6,       
+03|7, 8           
+04|~              
+05|~              
+06|~              
+07|~              
+08|~              
+09|<e] [+] 3,4 All
+10|               
+
+--|---------|-----
+01|000000000000000
+02|000000000000000
+03|000000000000000
+04|111111111111111
+05|111111111111111
+06|111111111111111
+07|111111111111111
+08|111111111111111
+09|222222222222222
+10|333333333333333
diff --git a/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-3-words,-works b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-3-words,-works
new file mode 100644
index 00000000..c53364af
--- /dev/null
+++ b/tests/screenshots/tests-test_suggestion.lua---suggestion()---accept_word,-3-words,-works
@@ -0,0 +1,23 @@
+--|---------|-----
+01|1, 2, 3,       
+02|4, 5, 6,       
+03|7, 8, 9        
+04|~              
+05|~              
+06|~              
+07|~              
+08|~              
+09|<e] [+] 3,7 All
+10|               
+
+--|---------|-----
+01|000000000000000
+02|000000000000000
+03|000000000000000
+04|111111111111111
+05|111111111111111
+06|111111111111111
+07|111111111111111
+08|111111111111111
+09|222222222222222
+10|333333333333333
diff --git a/tests/test_suggestion.lua b/tests/test_suggestion.lua
index ecf7ce2a..5b8248d1 100644
--- a/tests/test_suggestion.lua
+++ b/tests/test_suggestion.lua
@@ -22,7 +22,7 @@ T["suggestion()"]["suggestion works"] = function()
   child.type_keys("i123", "<Esc>", "o456", "<Esc>", "o7")
   child.wait_for_suggestion()
 
-  reference_screenshot(child.get_screenshot())
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
 end
 
 T["suggestion()"]["auto_trigger is false, will not show ghost test"] = function()
@@ -32,7 +32,7 @@ T["suggestion()"]["auto_trigger is false, will not show ghost test"] = function(
   vim.loop.sleep(3000)
   child.lua("vim.wait(0)")
 
-  reference_screenshot(child.get_screenshot())
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
 end
 
 T["suggestion()"]["accept keymap to trigger sugestion"] = function()
@@ -42,7 +42,7 @@ T["suggestion()"]["accept keymap to trigger sugestion"] = function()
   child.type_keys("i123", "<Esc>", "o456", "<Esc>", "o7", "<Tab>")
   child.wait_for_suggestion()
 
-  reference_screenshot(child.get_screenshot())
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
 end
 
 T["suggestion()"]["accept keymap, no suggestion, execute normal keystroke"] = function()
@@ -53,7 +53,118 @@ T["suggestion()"]["accept keymap, no suggestion, execute normal keystroke"] = fu
   child.configure_copilot()
   child.type_keys("i123", "<Esc>", "o456", "<Esc>", "o7", "<Tab>")
 
-  reference_screenshot(child.get_screenshot())
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
+end
+
+T["suggestion()"]["accept_word, 1 word, works"] = function()
+  child.o.lines, child.o.columns = 10, 15
+  child.config.suggestion = child.config.suggestion .. "auto_trigger = true," .. "keymap = { accept_word = '<C-e>' },"
+  child.configure_copilot()
+  child.type_keys("i1, 2, 3,", "<Esc>", "o4, 5, 6,", "<Esc>", "o7, ")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<Esc>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
+end
+
+T["suggestion()"]["accept_word, 3 words, works"] = function()
+  child.o.lines, child.o.columns = 10, 15
+  child.config.suggestion = child.config.suggestion .. "auto_trigger = true," .. "keymap = { accept_word = '<C-e>' },"
+  child.configure_copilot()
+  child.type_keys("i1, 2, 3,", "<Esc>", "o4, 5, 6,", "<Esc>", "o7, ")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<C-e>", "<C-e>", "<Esc>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
+end
+
+-- - accept_word, 1 word then next
+-- - accept_word, 1 word then prev
+
+T["suggestion()"]["accept_word, 1 word, then dismiss"] = function()
+  child.o.lines, child.o.columns = 10, 15
+  child.config.suggestion = child.config.suggestion
+    .. "auto_trigger = true,"
+    .. "keymap = { accept_word = '<C-e>', dismiss = '<Tab>' },"
+  child.configure_copilot()
+  child.type_keys("i1, 2, 3,", "<Esc>", "o4, 5, 6,", "<Esc>", "o7, ")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<Tab>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
+end
+
+T["suggestion()"]["accept_word, 1 word, then accept"] = function()
+  child.o.lines, child.o.columns = 10, 15
+  child.config.suggestion = child.config.suggestion
+    .. "auto_trigger = true,"
+    .. "keymap = { accept_word = '<C-e>', accept = '<Tab>' },"
+  child.configure_copilot()
+  child.type_keys("i1, 2, 3,", "<Esc>", "o4, 5, 6,", "<Esc>", "o7, ")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<Tab>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 9, 10 } })
+end
+
+T["suggestion()"]["accept_line, 1 line, works"] = function()
+  child.o.lines, child.o.columns = 30, 15
+  child.config.suggestion = child.config.suggestion .. "auto_trigger = true," .. "keymap = { accept_line = '<C-e>' },"
+  child.configure_copilot()
+  child.type_keys("i{", "<Esc>o", "  1,2,3", "<Esc>o", "4,5,6", "<Esc>o", "7,8,9", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o", "  10,11,12", "<Esc>", "o13,14,15", "<Esc>", "o16,17,18", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<Esc>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 29, 30 } })
+end
+
+T["suggestion()"]["accept_line, 3 lines, works"] = function()
+  child.o.lines, child.o.columns = 50, 15
+  child.config.suggestion = child.config.suggestion .. "auto_trigger = true," .. "keymap = { accept_line = '<C-e>' },"
+  child.configure_copilot()
+  child.type_keys("i{", "<Esc>o", "  1,2,3", "<Esc>o", "4,5,6", "<Esc>o", "7,8,9", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o", "  10,11,12", "<Esc>", "o13,14,15", "<Esc>", "o16,17,18", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<C-e>", "<C-e>", "<Esc>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 49, 50 } })
+end
+
+-- - accept_line, 1 line then next
+-- - accept_line, 1 line then prev
+
+T["suggestion()"]["accept_line, 1 line, then dismiss"] = function()
+  child.o.lines, child.o.columns = 30, 15
+  child.config.suggestion = child.config.suggestion
+    .. "auto_trigger = true,"
+    .. "keymap = { accept_line = '<C-e>', dismiss = '<Tab>' },"
+  child.configure_copilot()
+  child.type_keys("i{", "<Esc>o", "  1,2,3", "<Esc>o", "4,5,6", "<Esc>o", "7,8,9", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o", "  10,11,12", "<Esc>", "o13,14,15", "<Esc>", "o16,17,18", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<Tab>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 29, 30 } })
+end
+
+T["suggestion()"]["accept_line, 1 line, then accept"] = function()
+  child.o.lines, child.o.columns = 50, 40
+  child.config.suggestion = child.config.suggestion
+    .. "auto_trigger = true,"
+    .. "keymap = { accept_line = '<C-e>', accept = '<Tab>' },"
+  child.configure_copilot()
+  child.type_keys("i# Numbers in a 3x3 grid, up to 63", "<Esc>")
+  child.type_keys("o{", "<Esc>o", "  1,2,3", "<Esc>o", "4,5,6", "<Esc>o", "7,8,9", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o", "  10,11,12", "<Esc>", "o13,14,15", "<Esc>", "o16,17,18", "<Esc>o<bs>", "}", "<Esc>")
+  child.type_keys("o{", "<Esc>o")
+  child.wait_for_suggestion()
+  child.type_keys("<C-e>", "<Tab>")
+
+  reference_screenshot(child.get_screenshot(), nil, { ignore_lines = { 49, 50 } })
 end
 
 return T