Special case for dicts assigned to function args

Author: AndrewRadev

autoload/sj/python.vim

@@ -102,7 +102,7 @@ function! sj#python#SplitTuple()
 endfunction
 
 function! sj#python#SplitArgs()
-  if search('\%#[[:keyword:].]\+(', 'e', line('.'))
+  if search('\%#[[:keyword:].=]\+(', 'e', line('.'))
     return sj#python#SplitTuple()
   endif
 endfunction
@@ -112,7 +112,7 @@ function! sj#python#JoinTuple()
 endfunction
 
 function! sj#python#JoinArgs()
-  if search('\%#[[:keyword:].]\+(', 'e', line('.'))
+  if search('\%#[[:keyword:].=]\+(', 'e', line('.'))
     return sj#python#JoinTuple()
   endif
 endfunction

spec/plugin/python_spec.rb

@@ -47,6 +47,22 @@
     assert_file_contents "spam = {spam: [1, 2, 3], 'spam, spam': 'eggs'}"
   end
 
+  specify "a dictionary passed to a named variable" do
+    set_file_contents "function_call(spam=dict(eggs=[1, 2], ham=[3, 4])))"
+
+    vim.search 'spam='
+    split
+
+    assert_file_contents <<~EOF
+      function_call(spam=dict(eggs=[1, 2],
+                              ham=[3, 4])))
+    EOF
+
+    join
+
+    assert_file_contents "function_call(spam=dict(eggs=[1, 2], ham=[3, 4])))"
+  end
+
   specify "lists" do
     set_file_contents 'spam = [1, [2, 3], 4]'