merge: make toggle_math star aware

refer: #3213

Author: lervag

autoload/vimtex/env.vim

@@ -365,6 +365,7 @@ function! vimtex#env#toggle_math() abort
 
   let l:current = get(l:open, 'name', l:open.match)
   let l:target = get(g:vimtex_env_toggle_math_map, l:current, '$')
+        \  .. get(l:open, 'starred', 0) ? '*' : ''
 
   call vimtex#env#change(l:open, l:close, l:target)
 endfunction

test/test-env/test-toggle-math.tex

@@ -14,6 +14,10 @@
   1+1=2
 \end{equation}
 
+\begin{equation*}
+  1+1=2
+\end{equation*}
+
 World
 
 \begin{equation}

test/test-env/test-toggle-math.vim

@@ -7,15 +7,15 @@ nnoremap q :qall!<cr>
 
 silent edit test-toggle-math.tex
 
-normal! 20G
+normal! 24G
 call vimtex#env#toggle_math()
 call assert_equal([
       \ 'World',
       \ '',
       \ '$f(x) = 1 + e^x$',
       \ '',
       \ '\end{document}',
-      \], getline(17, 21))
+      \], getline(21, 25))
 
 normal! 7G
 call vimtex#env#toggle_math()
@@ -24,4 +24,24 @@ call assert_equal([
       \ '\end{proof}',
       \], getline(5, 6))
 
+let g:vimtex_env_toggle_math_map = {
+      \ 'equation': 'align',
+      \}
+
+normal! 11G
+call vimtex#env#toggle_math()
+call assert_equal([
+      \ '\begin{align}',
+      \ '  1+1=2',
+      \ '\end{align}',
+      \], getline(10, 12))
+
+normal! 15G
+call vimtex#env#toggle_math()
+call assert_equal([
+      \ '\begin{align*}',
+      \ '  1+1=2',
+      \ '\end{align*}',
+      \], getline(14, 16))
+
 call vimtex#test#finished()