]> projects.mako.cc - selectricity/blob - lib/gruff-0.2.8/test/test_net.rb
Ordered pref tables so that the smallest margin of vicotry appears first and the...
[selectricity] / lib / gruff-0.2.8 / test / test_net.rb
1 #!/usr/bin/ruby
2
3 require File.dirname(__FILE__) + "/gruff_test_case"
4
5 class TestGruffNet < GruffTestCase
6
7   def setup
8     @datasets = [
9       [:Jimmy, [25, 36, 86, 39, 25, 31, 79, 88]],
10       [:Charles, [80, 54, 67, 54, 68, 70, 90, 95]],
11       [:Julie, [22, 29, 35, 38, 36, 40, 46, 57]],
12       [:Jane, [95, 95, 95, 90, 85, 80, 88, 100]],
13       [:Philip, [90, 34, 23, 12, 78, 89, 98, 88]],
14       ["Arthur", [5, 10, 13, 11, 6, 16, 22, 32]],
15       ]
16
17     @sample_labels = {
18         0 => '5/6', 
19         1 => '5/15', 
20         2 => '5/24', 
21         3 => '5/30', 
22         4 => '6/4', 
23         5 => '6/12', 
24         6 => '6/21', 
25         7 => '6/28', 
26       }      
27   end
28     
29   def test_net_small_values    
30     @datasets = [
31       [:small, [0.1, 0.14356, 0.0, 0.5674839, 0.456]],
32       [:small2, [0.2, 0.3, 0.1, 0.05, 0.9]]
33       ]
34
35     g = Gruff::Net.new
36     g.title = "Small Values Net Graph Test"
37     @datasets.each do |data|
38       g.data(data[0], data[1])
39     end
40     g.write("test/output/net_small.png")
41
42     g = Gruff::Net.new(400)
43     g.title = "Small Values Net Graph Test 400px"
44     @datasets.each do |data|
45       g.data(data[0], data[1])
46     end
47     g.write("test/output/net_small_small.png")
48   end
49
50   def test_net_starts_with_zero
51     @datasets = [
52       [:first0, [0, 5, 10, 8, 18]],
53       [:normal, [1, 2, 3, 4, 5]]
54       ]
55
56     g = Gruff::Net.new
57     g.title = "Small Values Net Graph Test"
58     @datasets.each do |data|
59       g.data(data[0], data[1])
60     end
61     g.write("test/output/net_small_zero.png")
62
63     g = Gruff::Net.new(400)
64     g.title = "Small Values Net Graph Test 400px"
65     @datasets.each do |data|
66       g.data(data[0], data[1])
67     end
68     g.write("test/output/net_small_small_zero.png")
69   end
70
71     
72   def test_net_large_values    
73     @datasets = [
74       [:large, [100_005, 35_000, 28_000, 27_000]],
75       [:large2, [35_000, 28_000, 27_000, 100_005]],
76       [:large3, [28_000, 27_000, 100_005, 35_000]],
77       [:large4, [1_238, 39_092, 27_938, 48_876]]
78       ]
79
80     g = Gruff::Net.new
81     g.title = "Very Large Values Net Graph Test"
82     @datasets.each do |data|
83       g.data(data[0], data[1])
84     end
85
86     g.write("test/output/net_large.png")
87   end
88     
89   def test_many_datapoints
90     g = Gruff::Net.new
91     g.title = "Many Multi-Net Graph Test"
92     g.labels = {
93       0 => 'June', 
94       10 => 'July', 
95       30 => 'August', 
96       50 => 'September', 
97     }
98     g.data('many points', (0..50).collect {|i| rand(100) })
99
100     # Default theme
101     g.write("test/output/net_many.png")
102   end
103
104
105   def test_similar_high_end_values
106     g = Gruff::Net.new
107     g.title = "Similar High End Values Test"
108     g.data('similar points', %w(29.43 29.459 29.498 29.53 29.548 29.589 29.619 29.66 29.689 29.849 29.878 29.74 29.769 29.79 29.808 29.828).collect {|i| i.to_f} )
109
110     # Default theme
111     g.write("test/output/net_similar_high_end_values.png")    
112   end
113
114   def test_many_nets_graph_small
115     g = Gruff::Net.new(400)
116     g.title = "Many Values Net Test 400px"
117     g.labels = {
118       0 => '5/6', 
119       10 => '5/15', 
120       20 => '5/24', 
121       30 => '5/30', 
122       40 => '6/4', 
123       50 => '6/16'
124     }
125     %w{jimmy jane philip arthur julie bert}.each do |student_name|
126       g.data(student_name, (0..50).collect { |i| rand 100 })
127     end
128
129     # Default theme
130     g.write("test/output/net_many_nets_small.png")
131   end
132
133   def test_dots_graph_tiny
134     g = Gruff::Net.new(300)
135     g.title = "Dots Test 300px"
136     g.labels = {
137       0 => '5/6', 
138       10 => '5/15', 
139       20 => '5/24', 
140       30 => '5/30', 
141       40 => '6/4', 
142       50 => '6/16'
143     }
144     %w{jimmy jane philip arthur julie bert}.each do |student_name|
145       g.data(student_name, (0..50).collect { |i| rand 100 })
146     end
147
148     # Default theme
149     g.write("test/output/net_dots_tiny.png")
150   end
151
152   def test_no_data
153     g = Gruff::Net.new(400)
154     g.title = "No Data"
155     # Default theme
156     g.write("test/output/net_no_data.png")
157     
158     g = Gruff::Net.new(400)
159     g.title = "No Data Title"
160     g.no_data_message = 'There is no data'
161     g.write("test/output/net_no_data_msg.png")
162   end
163
164
165   def test_all_zeros
166     g = Gruff::Net.new(400)
167     g.title = "All Zeros"
168
169     g.data(:gus, [0,0,0,0])
170
171     # Default theme
172     g.write("test/output/net_no_data_other.png")    
173   end
174
175   def test_no_title
176     g = Gruff::Net.new(400)
177     g.labels = @sample_labels
178     @datasets.each do |data|
179       g.data(data[0], data[1])
180     end
181
182     g.write("test/output/net_no_title.png")
183   end
184
185   def test_no_net_markers
186     g = setup_basic_graph(400)
187     g.title = "No Net Markers"
188     g.hide_line_markers = true
189     g.write("test/output/net_no_net_markers.png")    
190   end
191
192   def test_no_legend
193     g = setup_basic_graph(400)
194     g.title = "No Legend"
195     g.hide_legend = true
196     g.write("test/output/net_no_legend.png")
197   end
198
199   def test_nothing_but_the_graph
200     g = setup_basic_graph(400)
201     g.title = "THIS TITLE SHOULD NOT DISPLAY!!!"
202     g.hide_line_markers = true
203     g.hide_legend = true
204     g.hide_title = true
205     g.write("test/output/net_nothing_but_the_graph.png")    
206   end
207
208   def test_wide_graph
209     g = setup_basic_graph('800x400')
210     g.title = "Wide Graph"
211     g.write("test/output/net_wide_graph.png")    
212
213     g = setup_basic_graph('400x200')
214     g.title = "Wide Graph Small"
215     g.write("test/output/net_wide_graph_small.png")
216   end
217
218 protected
219
220   def setup_basic_graph(size=800)
221     g = Gruff::Net.new(size)
222     g.title = "My Graph Title"
223     g.labels = @sample_labels
224     @datasets.each do |data|
225       g.data(data[0], data[1])
226     end
227     return g
228   end
229   
230 end

Benjamin Mako Hill || Want to submit a patch?