From e0be8bfc4ff68e6eb26662fe3a1881cba225a44e Mon Sep 17 00:00:00 2001
From: Ivan Smirnov <aldanor@users.noreply.github.com>
Date: Mon, 21 Aug 2017 16:59:17 +0100
Subject: [PATCH] Indent next line if prev line is empty (+tests) (#88)

---
 indent/python.vim          |  7 +++++++
 spec/indent/indent_spec.rb | 14 ++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/indent/python.vim b/indent/python.vim
index 19b649e..347e60d 100644
--- a/indent/python.vim
+++ b/indent/python.vim
@@ -301,6 +301,13 @@ function! s:indent_like_previous_line(lnum)
 
     let empty = getline(a:lnum) =~# '^\s*$'
 
+    " Current and prev line are empty, next is not -> indent like next.
+    if empty && a:lnum > 1 &&
+          \ (getline(a:lnum - 1) =~# '^\s*$') &&
+          \ !(getline(a:lnum + 1) =~# '^\s*$')
+      return indent(a:lnum + 1)
+    endif
+
     " If the previous statement was a stop-execution statement or a pass
     if getline(start) =~# s:stop_statement
         " Remove one level of indentation if the user hasn't already dedented
diff --git a/spec/indent/indent_spec.rb b/spec/indent/indent_spec.rb
index 85bc04b..23c1629 100644
--- a/spec/indent/indent_spec.rb
+++ b/spec/indent/indent_spec.rb
@@ -168,6 +168,20 @@ shared_examples_for "vim" do
     end
   end
 
+  describe "when an empty line is after empty line / before non-empty" do
+    it "is indented like the next line" do
+      vim.feedkeys 'idef a():\<CR>1\<CR>\<CR>\<CR>2\<ESC><<kcc'
+      indent.should == 0
+    end
+  end
+
+  describe "when an empty line is after empty line / before non-empty (nested)" do
+    it "is indented like the next line" do
+      vim.feedkeys 'idef a():\<CR>1\<CR>\<CR>\<CR>\<ESC>0i\<TAB>2\<ESC>kcc'
+      indent.should == shiftwidth
+    end
+  end
+
   describe "when line is empty inside a block following multi-line statement" do
     it "is indented like the previous line" do
       vim.feedkeys 'idef a():\<CR>x = (1 +\<CR>2)\<CR>\<CR>y\<ESC>kcc'
-- 
2.39.2