1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#!/usr/bin/env ruby
# encoding: UTF-8
require File.expand_path('../test_helper', __FILE__)
require 'stringio'
require 'fileutils'
require 'tmpdir'
require_relative 'prime'
# -- Tests ----
class PrinterGraphTest < TestCase
def setup
# WALL_TIME so we can use sleep in our test and get same measurements on linux and windows
RubyProf::measure_mode = RubyProf::WALL_TIME
@result = RubyProf.profile 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
|
