RubyGems - ruby-prof - Versions diffs - 1.7.1 → 2.0.0 - Mend

ruby-prof 1.7.1 → 2.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (134) hide show

checksums.yaml +4 -4
data/{CHANGES → CHANGELOG.md} +118 -176
data/README.md +5 -5
data/bin/ruby-prof +1 -4
data/docs/advanced-usage.md +132 -0
data/docs/alternatives.md +98 -0
data/docs/architecture.md +122 -0
data/docs/best-practices.md +27 -0
data/docs/getting-started.md +130 -0
data/docs/history.md +11 -0
data/docs/index.md +45 -0
data/docs/profiling-rails.md +64 -0
data/docs/public/examples/example.rb +33 -0
data/docs/public/examples/generate_reports.rb +92 -0
data/docs/public/examples/reports/call_info.txt +27 -0
data/docs/public/examples/reports/call_stack.html +835 -0
data/docs/public/examples/reports/callgrind.out +150 -0
data/docs/public/examples/reports/flame_graph.html +408 -0
data/docs/public/examples/reports/flat.txt +45 -0
data/docs/public/examples/reports/graph.dot +129 -0
data/docs/public/examples/reports/graph.html +1319 -0
data/docs/public/examples/reports/graph.txt +100 -0
data/docs/public/examples/reports/graphviz_viewer.html +1 -0
data/docs/public/images/call_stack.png +0 -0
data/docs/public/images/class_diagram.png +0 -0
data/docs/public/images/dot_printer.png +0 -0
data/docs/public/images/flame_graph.png +0 -0
data/docs/public/images/flat.png +0 -0
data/docs/public/images/graph.png +0 -0
data/docs/public/images/graph_html.png +0 -0
data/docs/public/images/ruby-prof-logo.svg +1 -0
data/docs/reports.md +150 -0
data/docs/stylesheets/extra.css +80 -0
data/ext/ruby_prof/extconf.rb +23 -22
data/ext/ruby_prof/rp_allocation.c +0 -15
data/ext/ruby_prof/rp_allocation.h +29 -33
data/ext/ruby_prof/rp_call_tree.c +3 -0
data/ext/ruby_prof/rp_call_tree.h +1 -4
data/ext/ruby_prof/rp_call_trees.c +296 -296
data/ext/ruby_prof/rp_call_trees.h +25 -28
data/ext/ruby_prof/rp_measure_allocations.c +47 -47
data/ext/ruby_prof/rp_measure_process_time.c +64 -66
data/ext/ruby_prof/rp_measure_wall_time.c +52 -64
data/ext/ruby_prof/rp_measurement.c +0 -5
data/ext/ruby_prof/rp_measurement.h +49 -53
data/ext/ruby_prof/rp_method.c +554 -551
data/ext/ruby_prof/rp_method.h +1 -4
data/ext/ruby_prof/rp_profile.c +1 -1
data/ext/ruby_prof/rp_profile.h +1 -5
data/ext/ruby_prof/rp_stack.c +212 -212
data/ext/ruby_prof/rp_stack.h +50 -53
data/ext/ruby_prof/rp_thread.h +1 -4
data/ext/ruby_prof/ruby_prof.c +50 -50
data/ext/ruby_prof/ruby_prof.h +4 -6
data/ext/ruby_prof/vc/ruby_prof.vcxproj +7 -8
data/lib/ruby-prof/assets/call_stack_printer.html.erb +746 -711
data/lib/ruby-prof/assets/flame_graph_printer.html.erb +412 -0
data/lib/ruby-prof/assets/graph_printer.html.erb +355 -355
data/lib/ruby-prof/call_tree.rb +57 -57
data/lib/ruby-prof/call_tree_visitor.rb +36 -36
data/lib/ruby-prof/exclude_common_methods.rb +204 -204
data/lib/ruby-prof/measurement.rb +17 -17
data/lib/ruby-prof/printers/abstract_printer.rb +142 -138
data/lib/ruby-prof/printers/call_info_printer.rb +53 -53
data/lib/ruby-prof/printers/call_stack_printer.rb +168 -180
data/lib/ruby-prof/printers/call_tree_printer.rb +132 -145
data/lib/ruby-prof/printers/dot_printer.rb +177 -132
data/lib/ruby-prof/printers/flame_graph_printer.rb +79 -0
data/lib/ruby-prof/printers/flat_printer.rb +52 -52
data/lib/ruby-prof/printers/graph_html_printer.rb +62 -63
data/lib/ruby-prof/printers/graph_printer.rb +112 -113
data/lib/ruby-prof/printers/multi_printer.rb +134 -127
data/lib/ruby-prof/profile.rb +13 -0
data/lib/ruby-prof/rack.rb +114 -105
data/lib/ruby-prof/task.rb +147 -147
data/lib/ruby-prof/thread.rb +20 -20
data/lib/ruby-prof/version.rb +3 -3
data/lib/ruby-prof.rb +50 -52
data/lib/unprof.rb +10 -10
data/ruby-prof.gemspec +66 -65
data/test/abstract_printer_test.rb +25 -27
data/test/alias_test.rb +203 -117
data/test/call_tree_builder.rb +126 -126
data/test/call_tree_visitor_test.rb +27 -27
data/test/call_trees_test.rb +66 -66
data/test/duplicate_names_test.rb +32 -32
data/test/dynamic_method_test.rb +50 -62
data/test/enumerable_test.rb +23 -21
data/test/exceptions_test.rb +24 -24
data/test/exclude_methods_test.rb +363 -257
data/test/exclude_threads_test.rb +48 -48
data/test/fiber_test.rb +195 -195
data/test/gc_test.rb +104 -102
data/test/inverse_call_tree_test.rb +174 -174
data/test/line_number_test.rb +563 -289
data/test/marshal_test.rb +144 -145
data/test/measure_allocations.rb +26 -26
data/test/measure_allocations_test.rb +1511 -1081
data/test/measure_process_time_test.rb +3286 -2477
data/test/measure_times.rb +56 -56
data/test/measure_wall_time_test.rb +773 -568
data/test/measurement_test.rb +82 -82
data/test/merge_test.rb +146 -146
data/test/method_info_test.rb +100 -95
data/test/multi_printer_test.rb +52 -66
data/test/no_method_class_test.rb +15 -15
data/test/pause_resume_test.rb +171 -171
data/test/prime.rb +54 -54
data/test/prime_script.rb +5 -5
data/test/printer_call_stack_test.rb +28 -27
data/test/printer_call_tree_test.rb +30 -30
data/test/printer_flame_graph_test.rb +82 -0
data/test/printer_flat_test.rb +99 -99
data/test/printer_graph_html_test.rb +62 -59
data/test/printer_graph_test.rb +42 -40
data/test/printers_test.rb +162 -135
data/test/printing_recursive_graph_test.rb +81 -81
data/test/profile_test.rb +101 -101
data/test/rack_test.rb +103 -93
data/test/recursive_test.rb +796 -622
data/test/scheduler.rb +4 -0
data/test/singleton_test.rb +39 -38
data/test/stack_printer_test.rb +61 -61
data/test/start_stop_test.rb +106 -106
data/test/test_helper.rb +24 -20
data/test/thread_test.rb +229 -231
data/test/unique_call_path_test.rb +123 -136
data/test/yarv_test.rb +56 -60
metadata +68 -16
data/ext/ruby_prof/rp_measure_memory.c +0 -46
data/lib/ruby-prof/compatibility.rb +0 -113
data/test/compatibility_test.rb +0 -49
data/test/crash2.rb +0 -144
data/test/measure_memory_test.rb +0 -1456

data/test/printer_call_tree_test.rb CHANGED Viewed

@@ -1,30 +1,30 @@
-#!/usr/bin/env ruby
-# encoding: UTF-8
-require File.expand_path('../test_helper', __FILE__)
-require 'fileutils'
-require 'tmpdir'
-require_relative 'prime'
-# --  Tests ----
-class PrinterCallTreeTest < TestCase
-  def setup
-    super
-    # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
-    @result = RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
-      run_primes(1000, 5000)
-    end
-  end
-  def test_call_tree_string
-    printer = RubyProf::CallTreePrinter.new(@result)
-    printer.print(:path => Dir.tmpdir)
-    main_output_file_name = File.join(Dir.tmpdir, "callgrind.out.#{$$}")
-    assert(File.exist?(main_output_file_name))
-    output = File.read(main_output_file_name)
-    assert_match(/fn=Object::find_primes/i, output)
-    assert_match(/events: wall_time/i, output)
-    refute_match(/d\d\d\d\d\d/, output) # old bug looked [in error] like Object::run_primes(d5833116)
-  end
-end
+#!/usr/bin/env ruby
+# encoding: UTF-8
+require File.expand_path('../test_helper', __FILE__)
+require 'fileutils'
+require 'tmpdir'
+require_relative 'prime'
+# --  Tests ----
+class PrinterCallTreeTest < TestCase
+  def setup
+    super
+    # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
+    @result = RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
+      run_primes(1000, 5000)
+    end
+  end
+  def test_call_tree_string
+    printer = RubyProf::CallTreePrinter.new(@result)
+    printer.print(path: Dir.tmpdir)
+    main_output_file_name = File.join(Dir.tmpdir, "callgrind.out.#{$$}")
+    assert(File.exist?(main_output_file_name))
+    output = File.read(main_output_file_name)
+    assert_match(/fn=Object::find_primes/i, output)
+    assert_match(/events: wall_time/i, output)
+    refute_match(/d\d\d\d\d\d/, output) # old bug looked [in error] like Object::run_primes(d5833116)
+  end
+end

data/test/printer_flame_graph_test.rb ADDED Viewed

@@ -0,0 +1,82 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+require File.expand_path('../test_helper', __FILE__)
+require 'fileutils'
+require 'stringio'
+require 'tmpdir'
+require_relative 'prime'
+# --  Tests ----
+class PrinterFlameGraphTest < TestCase
+  def setup
+    super
+    # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
+    @result = RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
+      run_primes(1000, 5000)
+    end
+  end
+  def test_flame_graph_string
+    output = StringIO.new
+    printer = RubyProf::FlameGraphPrinter.new(@result)
+    printer.print(output)
+    assert_match(/<!DOCTYPE html>/i, output.string)
+    assert_match(/flame-svg/, output.string)
+    assert_match(/Object#run_primes/i, output.string)
+  end
+  def test_flame_graph_stringio
+    output = StringIO.new
+    printer = RubyProf::FlameGraphPrinter.new(@result)
+    printer.print(output)
+    result = output.string
+    assert_match(/<!DOCTYPE html>/i, result)
+    assert_match(/flame-svg/, result)
+    assert_match(/Object#run_primes/i, result)
+  end
+  def test_flame_graph_contains_svg_elements
+    output = StringIO.new
+    printer = RubyProf::FlameGraphPrinter.new(@result)
+    printer.print(output)
+    assert_match(/<svg/, output.string)
+    assert_match(/renderNode/, output.string)
+  end
+  def test_flame_graph_contains_json_data
+    output = StringIO.new
+    printer = RubyProf::FlameGraphPrinter.new(@result)
+    printer.print(output)
+    # The template embeds thread data as JSON
+    assert_match(/"name"/, output.string)
+    assert_match(/"value"/, output.string)
+    assert_match(/"children"/, output.string)
+  end
+  def test_flame_graph_custom_title
+    output = StringIO.new
+    printer = RubyProf::FlameGraphPrinter.new(@result)
+    printer.print(output, title: "Custom Flame Graph")
+    assert_match(/Custom Flame Graph/, output.string)
+  end
+  def test_flame_graph_file_output
+    Dir.mktmpdir do |dir|
+      path = File.join(dir, "flame_graph.html")
+      File.open(path, "wb") do |file|
+        printer = RubyProf::FlameGraphPrinter.new(@result)
+        printer.print(file)
+      end
+      content = File.read(path)
+      assert_match(/<!DOCTYPE html>/i, content)
+      assert_match(/Object#run_primes/i, content)
+    end
+  end
+end

data/test/printer_flat_test.rb CHANGED Viewed

@@ -1,99 +1,99 @@
-#!/usr/bin/env ruby
-# encoding: UTF-8
-require File.expand_path('../test_helper', __FILE__)
-require 'fileutils'
-require 'stringio'
-require 'tmpdir'
-require_relative 'prime'
-# --  Tests ----
-class PrinterFlatTest < TestCase
-  def run_profile
-    RubyProf::Profile.profile(:measure_mode => RubyProf::WALL_TIME) do
-      run_primes(1000, 5000)
-    end
-  end
-  def flat_output_nth_column_values(output, n)
-    only_method_calls = output.split("\n").select { |line| line =~ /^\s+\d+/ }
-    only_method_calls.collect { |line| line.split(/\s+/)[n] }
-  end
-  def helper_test_flat_string(klass)
-    output = StringIO.new
-    printer = klass.new(self.run_profile)
-    printer.print(output)
-    assert_match(/Thread ID: -?\d+/i, output.string)
-    assert_match(/Fiber ID: -?\d+/i, output.string)
-    assert_match(/Total: \d+\.\d+/i, output.string)
-    assert_match(/Object#run_primes/i, output.string)
-    output.string
-  end
-  def assert_sorted(array)
-    array = array.map(&:to_f) # allow for > 10s times to sort right, since lexographically 4.0 > 10.0
-    assert_equal(array, array.sort.reverse)
-  end
-  def test_flat_string
-    output = helper_test_flat_string(RubyProf::FlatPrinter)
-    assert_match(/prime.rb/, output)
-  end
-  def test_flat_result_sorting_by_self_time_is_default
-    printer = RubyProf::FlatPrinter.new(self.run_profile)
-    output = StringIO.new
-    printer.print(output)
-    self_times = flat_output_nth_column_values(output.string, 3)
-    assert_sorted self_times
-  end
-  def test_flat_result_sorting
-    printer = RubyProf::FlatPrinter.new(self.run_profile)
-    sort_method_with_column_number = {:total_time => 2, :self_time => 3, :wait_time => 4, :children_time => 5}
-    sort_method_with_column_number.each_pair do |sort_method, n|
-      output = StringIO.new
-      printer.print(output, :sort_method => sort_method)
-      times = flat_output_nth_column_values(output.string, n)
-      assert_sorted(times)
-    end
-  end
-  def test_flat_result_max_percent
-    printer = RubyProf::FlatPrinter.new(self.run_profile)
-    output = StringIO.new
-    printer.print(output, max_percent: 1)
-    self_percents = flat_output_nth_column_values(output.string, 1).map(&:to_f)
-    assert self_percents.max < 1
-  end
-  def test_flat_result_filter_by_total_time
-    printer = RubyProf::FlatPrinter.new(self.run_profile)
-    output = StringIO.new
-    printer.print(output, filter_by: :total_time, min_percent: 50)
-    total_times = flat_output_nth_column_values(output.string, 2).map(&:to_f)
-    assert (total_times.min / total_times.max) >= 0.5
-  end
-  def test_flat_result_filter_by_self_time
-    printer = RubyProf::FlatPrinter.new(self.run_profile)
-    output = StringIO.new
-    printer.print(output, filter_by: :self_time, min_percent: 0.1)
-    self_percents = flat_output_nth_column_values(output.string, 1).map(&:to_f)
-    assert self_percents.min >= 0.1
-  end
-end
+#!/usr/bin/env ruby
+# encoding: UTF-8
+require File.expand_path('../test_helper', __FILE__)
+require 'fileutils'
+require 'stringio'
+require 'tmpdir'
+require_relative 'prime'
+# --  Tests ----
+class PrinterFlatTest < TestCase
+  def run_profile
+    RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
+      run_primes(1000, 5000)
+    end
+  end
+  def flat_output_nth_column_values(output, n)
+    only_method_calls = output.split("\n").select { |line| line =~ /^\s+\d+/ }
+    only_method_calls.collect { |line| line.split(/\s+/)[n] }
+  end
+  def helper_test_flat_string(klass)
+    output = StringIO.new
+    printer = klass.new(self.run_profile)
+    printer.print(output)
+    assert_match(/Thread ID: -?\d+/i, output.string)
+    assert_match(/Fiber ID: -?\d+/i, output.string)
+    assert_match(/Total: \d+\.\d+/i, output.string)
+    assert_match(/Object#run_primes/i, output.string)
+    output.string
+  end
+  def assert_sorted(array)
+    array = array.map(&:to_f) # allow for > 10s times to sort right, since lexographically 4.0 > 10.0
+    assert_equal(array, array.sort.reverse)
+  end
+  def test_flat_string
+    output = helper_test_flat_string(RubyProf::FlatPrinter)
+    assert_match(/prime.rb/, output)
+  end
+  def test_flat_result_sorting_by_self_time_is_default
+    printer = RubyProf::FlatPrinter.new(self.run_profile)
+    output = StringIO.new
+    printer.print(output)
+    self_times = flat_output_nth_column_values(output.string, 3)
+    assert_sorted self_times
+  end
+  def test_flat_result_sorting
+    printer = RubyProf::FlatPrinter.new(self.run_profile)
+    sort_method_with_column_number = {total_time: 2, self_time: 3, wait_time: 4, children_time: 5}
+    sort_method_with_column_number.each_pair do |sort_method, n|
+      output = StringIO.new
+      printer.print(output, sort_method: sort_method)
+      times = flat_output_nth_column_values(output.string, n)
+      assert_sorted(times)
+    end
+  end
+  def test_flat_result_max_percent
+    printer = RubyProf::FlatPrinter.new(self.run_profile)
+    output = StringIO.new
+    printer.print(output, max_percent: 1)
+    self_percents = flat_output_nth_column_values(output.string, 1).map(&:to_f)
+    assert self_percents.max < 1
+  end
+  def test_flat_result_filter_by_total_time
+    printer = RubyProf::FlatPrinter.new(self.run_profile)
+    output = StringIO.new
+    printer.print(output, filter_by: :total_time, min_percent: 50)
+    total_times = flat_output_nth_column_values(output.string, 2).map(&:to_f)
+    assert (total_times.min / total_times.max) >= 0.5
+  end
+  def test_flat_result_filter_by_self_time
+    printer = RubyProf::FlatPrinter.new(self.run_profile)
+    output = StringIO.new
+    printer.print(output, filter_by: :self_time, min_percent: 0.1)
+    self_percents = flat_output_nth_column_values(output.string, 1).map(&:to_f)
+    assert self_percents.min >= 0.1
+  end
+end

data/test/printer_graph_html_test.rb CHANGED Viewed

@@ -1,59 +1,62 @@
-#!/usr/bin/env ruby
-# encoding: UTF-8
-require File.expand_path('../test_helper', __FILE__)
-require 'fileutils'
-require 'tmpdir'
-require_relative 'prime'
-# --  Tests ----
-class PrinterGraphHtmlTest < TestCase
-  def setup
-    super
-    # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
-    @result = RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
-      run_primes(1000, 5000)
-    end
-  end
-  def graph_html_output_nth_column_values(output, n)
-    only_root_calls = output.split('<tr class="method">')
-    only_root_calls.delete_at(0)
-    only_root_calls.collect {|line| line.scan(/[\d\.]+/)[n - 1] }
-  end
-  def assert_sorted array
-    array = array.map{|n| n.to_f} # allow for > 10s times to sort right, since lexographically 4.0 > 10.0
-    assert_equal array, array.sort.reverse, "Array #{array.inspect} is not sorted"
-  end
-  def test_graph_html_string
-    output = ''
-    printer = RubyProf::GraphHtmlPrinter.new(@result)
-    printer.print(output)
-    assert_match(/<!DOCTYPE html>/i, output)
-    assert_match( %r{<th>Total</th>}i, output)
-    assert_match(/Object#run_primes/i, output)
-  end
-  def test_graph_html_result_sorting_by_total_time_is_default
-    printer = RubyProf::GraphHtmlPrinter.new(@result)
-    printer.print(output = '')
-    total_times = graph_html_output_nth_column_values(output, 3)
-    assert_sorted total_times
-  end
-  def test_graph_html_result_sorting
-    printer = RubyProf::GraphHtmlPrinter.new(@result)
-    sort_method_with_column_number = {:total_time => 3, :self_time => 4, :wait_time => 5, :children_time => 6}
-    sort_method_with_column_number.each_pair do |sort_method, n|
-      printer.print(output = '', :sort_method => sort_method)
-      times = graph_html_output_nth_column_values(output, n)
-      assert_sorted times
-    end
-  end
-end
+#!/usr/bin/env ruby
+# encoding: UTF-8
+require File.expand_path('../test_helper', __FILE__)
+require 'fileutils'
+require 'stringio'
+require 'tmpdir'
+require_relative 'prime'
+# --  Tests ----
+class PrinterGraphHtmlTest < TestCase
+  def setup
+    super
+    # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
+    @result = RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
+      run_primes(1000, 5000)
+    end
+  end
+  def graph_html_output_nth_column_values(output, n)
+    only_root_calls = output.split('<tr class="method">')
+    only_root_calls.delete_at(0)
+    only_root_calls.collect {|line| line.scan(/[\d\.]+/)[n - 1] }
+  end
+  def assert_sorted array
+    array = array.map{|n| n.to_f} # allow for > 10s times to sort right, since lexographically 4.0 > 10.0
+    assert_equal array, array.sort.reverse, "Array #{array.inspect} is not sorted"
+  end
+  def test_graph_html_string
+    output = StringIO.new
+    printer = RubyProf::GraphHtmlPrinter.new(@result)
+    printer.print(output)
+    assert_match(/<!DOCTYPE html>/i, output.string)
+    assert_match( %r{<th>Total</th>}i, output.string)
+    assert_match(/Object#run_primes/i, output.string)
+  end
+  def test_graph_html_result_sorting_by_total_time_is_default
+    printer = RubyProf::GraphHtmlPrinter.new(@result)
+    output = StringIO.new
+    printer.print(output)
+    total_times = graph_html_output_nth_column_values(output.string, 3)
+    assert_sorted total_times
+  end
+  def test_graph_html_result_sorting
+    printer = RubyProf::GraphHtmlPrinter.new(@result)
+    sort_method_with_column_number = {total_time: 3, self_time: 4, wait_time: 5, children_time: 6}
+    sort_method_with_column_number.each_pair do |sort_method, n|
+      output = StringIO.new
+      printer.print(output, sort_method: sort_method)
+      times = graph_html_output_nth_column_values(output.string, n)
+      assert_sorted times
+    end
+  end
+end

data/test/printer_graph_test.rb CHANGED Viewed

@@ -1,40 +1,42 @@
-#!/usr/bin/env ruby
-# encoding: UTF-8
-require File.expand_path('../test_helper', __FILE__)
-require 'fileutils'
-require 'tmpdir'
-require_relative 'prime'
-# --  Tests ----
-class PrinterGraphTest < TestCase
-  def setup
-    super
-    # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
-    @result = RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
-      run_primes(1000, 5000)
-    end
-  end
-  def graph_output_nth_column_values(output, n)
-    only_root_calls = output.split("\n").select { |line| line =~ /^ +[\d\.]+%/ }
-    only_root_calls.collect { |line| line.split(/ +/)[n] }
-  end
-  def assert_sorted(array)
-    array = array.map {|n| n.to_f} # allow for > 10s times to sort right, since lexographically 4.0 > 10.0
-    assert_equal(array, array.sort.reverse, "Array #{array.inspect} is not sorted")
-  end
-  def test_graph_results_sorting
-    printer = RubyProf::GraphPrinter.new(@result)
-    sort_method_with_column_number = {:total_time => 3, :self_time => 4, :wait_time => 5, :children_time => 6}
-    sort_method_with_column_number.each_pair do |sort_method, n|
-      printer.print(output = '', :sort_method => sort_method)
-      times = graph_output_nth_column_values(output, n)
-      assert_sorted times
-    end
-  end
-end
+#!/usr/bin/env ruby
+# encoding: UTF-8
+require File.expand_path('../test_helper', __FILE__)
+require 'fileutils'
+require 'stringio'
+require 'tmpdir'
+require_relative 'prime'
+# --  Tests ----
+class PrinterGraphTest < TestCase
+  def setup
+    super
+    # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
+    @result = RubyProf::Profile.profile(measure_mode: RubyProf::WALL_TIME) do
+      run_primes(1000, 5000)
+    end
+  end
+  def graph_output_nth_column_values(output, n)
+    only_root_calls = output.split("\n").select { |line| line =~ /^ +[\d\.]+%/ }
+    only_root_calls.collect { |line| line.split(/ +/)[n] }
+  end
+  def assert_sorted(array)
+    array = array.map {|n| n.to_f} # allow for > 10s times to sort right, since lexographically 4.0 > 10.0
+    assert_equal(array, array.sort.reverse, "Array #{array.inspect} is not sorted")
+  end
+  def test_graph_results_sorting
+    printer = RubyProf::GraphPrinter.new(@result)
+    sort_method_with_column_number = {total_time: 3, self_time: 4, wait_time: 5, children_time: 6}
+    sort_method_with_column_number.each_pair do |sort_method, n|
+      output = StringIO.new
+      printer.print(output, sort_method: sort_method)
+      times = graph_output_nth_column_values(output.string, n)
+      assert_sorted times
+    end
+  end
+end