module RuboCop::Cop::MinitestExplorationHelpers

def assertions(def_node)
  method_def = def_node.body
  return [] unless method_def
  send_nodes =
    if method_def.send_type?
      [method_def]
    else
      method_def.each_child_node(:send)
    end
  send_nodes.select { |send_node| assertion_method?(send_node) }
end