class_structure.rb in rubocop-1.9.0

- old
+ new

@@ -211,11 +211,16 @@
         # @return [String] with the key category or the `method_name` as string
         def find_category(node)
           name = node.method_name.to_s
           category, = categories.find { |_, names| names.include?(name) }
           key = category || name
-          visibility_key = "#{node_visibility(node)}_#{key}"
+          visibility_key =
+            if node.def_modifier?
+              "#{name}_methods"
+            else
+              "#{node_visibility(node)}_#{key}"
+            end
           expected_order.include?(visibility_key) ? visibility_key : key
         end
 
         def walk_over_nested_class_definition(class_node)
           class_elements(class_node).each do |node|
@@ -262,10 +267,10 @@
           HUMANIZED_NODE_TYPE[node.type] || node.type
         end
 
         def source_range_with_comment(node)
           begin_pos, end_pos =
-            if node.def_type?
+            if node.def_type? && !node.method?(:initialize) || node.send_type? && node.def_modifier?
               start_node = find_visibility_start(node) || node
               end_node = find_visibility_end(node) || node
               [begin_pos_with_comment(start_node),
                end_position_for(end_node) + 1]
             else