]> projects.mako.cc - selectricity-live/blob - lib/gruff-0.2.8/test/test_spider.rb
Merge jdong
[selectricity-live] / lib / gruff-0.2.8 / test / test_spider.rb
1 #!/usr/bin/ruby
2
3 require File.dirname(__FILE__) + "/gruff_test_case"
4
5 class TestGruffSpider < GruffTestCase
6
7   def setup
8     @datasets = [
9       [:Strength, [10]],
10       [:Dexterity, [16]],
11       [:Constitution, [12]],
12       [:Intelligence, [12]],
13       [:Wisdom, [10]],
14       ["Charisma", [16]],
15       ]
16
17 #     @datasets = [
18 #       [:Darren, [25]],
19 #       [:Chris, [80]],
20 #       [:Egbert, [22]],
21 #       [:Adam, [95]],
22 #       [:Bill, [90]],
23 #       ["Frank", [5]],
24 #       ["Zero", [0]],
25 #       ]
26   end
27   
28   def test_spider_graph
29     g = Gruff::Spider.new(20)
30     g.title = "Spider Graph Test"
31     @datasets.each do |data|
32       g.data(data[0], data[1])
33     end
34
35     # Default theme
36     g.write("test/output/spider_keynote.png")
37   end
38
39   def test_pie_graph_small
40     g = Gruff::Spider.new(20, 400)
41     g.title = "Visual Spider Graph Test Small"
42     @datasets.each do |data|
43       g.data(data[0], data[1])
44     end
45
46     # Default theme
47     g.write("test/output/spider_small.png")
48   end
49
50   def test_spider_graph_nearly_equal
51     g = Gruff::Spider.new(50)
52     g.title = "Spider Graph Nearly Equal"
53     
54     g.data(:Blake, [41])
55     g.data(:Aaron, [42])
56     g.data(:Grouch, [40])
57 #    g.data(:Snuffleupagus, [43])
58
59     g.write("test/output/spider_nearly_equal.png")
60   end
61
62   def test_pie_graph_equal
63     g = Gruff::Spider.new(50)
64     g.title = "Spider Graph Equal"
65         
66     g.data(:Bert, [41])
67     g.data(:Adam, [41])
68     g.data(:Joe, [41])
69
70     g.write("test/output/spider_equal.png")
71   end
72
73   def test_pie_graph_zero
74     g = Gruff::Spider.new(2)
75     g.title = "Pie Graph Two One Zero"
76         
77     g.data(:Bert, [0])
78     g.data(:Adam, [1])
79     g.data(:Sam,  [2])
80
81     g.write("test/output/spider_zero.png")
82   end
83
84   def test_wide
85     g = setup_basic_graph('800x400')
86     g.title = "Wide spider"
87     g.write("test/output/spider_wide.png")
88   end
89
90   def test_label_size
91     g = setup_basic_graph()
92     g.title = "Spider With Small Legend"
93     g.legend_font_size = 10
94     g.write("test/output/spider_legend.png")
95     
96     g = setup_basic_graph(400)
97     g.title = "Small spider With Small Legend"
98     g.legend_font_size = 10
99     g.write("test/output/spider_legend_small.png")
100   end
101  
102   def test_theme_37signals
103     g = Gruff::Spider.new(20)
104     g.title = "Spider Graph Test"
105     @datasets.each do |data|
106      g.data(data[0], data[1])
107     end
108
109     g.theme_37signals
110
111     # Default theme
112     g.write("test/output/spider_37signals.png")
113   end
114  
115   def test_no_axes
116     g = Gruff::Spider.new(20)
117     g.title = "Look ma, no axes"
118     g.hide_axes = true
119     @datasets.each do |data|
120      g.data(data[0], data[1])
121     end
122     g.write("test/output/spider_no_axes.png")
123
124   end
125  
126   def test_no_print
127     g = Gruff::Spider.new(20)
128     g.title = "Should not print"
129     g.hide_text = true
130     @datasets.each do |data|
131       g.data(data[0], data[1])
132     end
133     g.write("test/output/spider_no_print.png")
134   end
135  
136   def test_transparency
137     g = Gruff::Spider.new(20)
138     g.title = "Transparent background"
139     g.hide_text = true
140     g.transparent_background = true
141     g.hide_axes = true
142     @datasets.each do |data|
143       g.data(data[0], data[1])
144     end
145     g.write("test/output/spider_no_background.png")
146   end
147   
148   def test_overlay
149     g = Gruff::Spider.new(20)
150     g.title = "George (blue) vs Sarah (white)"
151     @datasets.each do |data|
152       g.data(data[0], data[1])
153     end
154     g.write("test/output/spider_overlay_1.png")
155     
156     g = Gruff::Spider.new(20)
157     g.title = "Transparent background"
158     g.hide_text = true
159     g.hide_axes = true
160     g.transparent_background = true
161     @datasets = [
162       [:Strength, [18]],
163       [:Dexterity, [10]],
164       [:Constitution, [18]],
165       [:Intelligence, [8]],
166       [:Wisdom, [14]],
167       ["Charisma", [4]],
168       ]   
169     @datasets.each do |data|
170       g.data(data[0], data[1])
171     end
172     g.marker_color = "#4F6EFF"
173     g.write("test/output/spider_overlay_2.png")
174   end  
175  
176   def test_lots_of_data
177     g = Gruff::Spider.new(10)
178     @datasets = [[:a, [1]], [:b, [5]], [:c, [3]], [:d, [9]], [:e, [4]],
179                  [:f, [7]], [:g, [0]], [:h, [4]], [:i, [6]], [:j, [0]],
180                  [:k, [4]], [:l, [8]]]
181    
182     @datasets.each do |data|
183       g.data(data[0], data[1])
184     end
185                  
186     g.title = "Sample Data"
187     g.write("test/output/spider_lots_of_data.png")
188   end
189   
190   def test_lots_of_data_with_large_names
191     g = Gruff::Spider.new(10)
192     @datasets = [[:anteaters, [1]], [:bulls, [5]], [:cats, [3]], [:dogs, [9]], [:elephants, [4]],
193                  [:frogs, [7]], [:giraffes, [0]], [:hamsters, [4]], [:iguanas, [6]], 
194                  [:jaguar, [0]], [:kangaroo, [4]], [:locust, [8]]]
195    
196     @datasets.each do |data|
197       g.data(data[0], data[1])
198     end
199                  
200     g.title = "Zoo Inventory"
201     g.write("test/output/spider_lots_of_data_normal_names.png")
202   end
203   
204
205 protected
206
207   def setup_basic_graph(size=800, max = 20)
208     g = Gruff::Spider.new(max, size)
209     g.title = "My Graph Title"
210     @datasets.each do |data|
211       g.data(data[0], data[1])
212     end
213     return g
214   end
215   
216 end

Benjamin Mako Hill || Want to submit a patch?