[fix] def formating was put before :

Author: Psycojoker

baron/formatting_grouper.py

@@ -141,6 +141,6 @@ def group_generator(sequence):
         # not really happy about this solution but that avoid a broken release
         if current[0] == "COLON" and iterator.show_next() and iterator.show_next()[0] == "COMMENT":
             comment = next(iterator)
-            current = (current[0], current[1], ((current[2]) if len(current) > 3 else []) + [comment])
+            current = (current[0], current[1], ((current[2]) if len(current) > 2 else []), ((current[3]) if len(current) > 3 else []) + [comment])
 
         yield current

tests/test_regression.py

@@ -1,4 +1,4 @@
-from baron import parse
+from baron import parse, dumps
 
 
 def test_regression_trailing_comment_after_colon():
@@ -7,3 +7,13 @@ def test_regression_trailing_comment_after_colon():
 
 def test_regression_trailing_comment_after_colon_no_space():
     assert parse("def a():# pouf\n    pass")
+
+
+def test_regression_trailing_comment_after_colon_dump():
+    code = "def a(): # pouf\n    pass\n"
+    assert dumps(parse(code)) == code
+
+
+def test_regression_trailing_comment_after_colon_no_space_dump():
+    code = "def a():# pouf\n    pass\n"
+    assert dumps(parse(code)) == code