~alf-rodrigo/cairoplot/trunk

« back to all changes in this revision

Viewing changes to trunk/tests.py

  • Committer: Rodrigo Moreira Araujo
  • Date: 2009-02-27 02:00:31 UTC
  • Revision ID: rodrigo@scrooge-20090227020031-u5tl2cpqo2a0quvb
tests.py
 - fixing but created after module rename.

Show diffs side-by-side

added added

removed removed

Lines of Context:
3
3
import random
4
4
import cairoplot
5
5
 
6
 
test_bar_plot = 0
7
 
test_scatter_plot = 0
8
 
test_donut_plot = 0
 
6
test_bar_plot = 1
 
7
test_scatter_plot = 1
 
8
test_donut_plot = 1
9
9
test_dot_line_plot = 1
10
 
test_function_plot = 0
11
 
test_gantt_chart = 0
12
 
test_pie_plot = 0
13
 
test_themes = 0
 
10
test_function_plot = 1
 
11
test_gantt_chart = 1
 
12
test_pie_plot = 1
 
13
test_themes = 1
14
14
 
15
15
if test_bar_plot:
16
16
    #Passing a dictionary
17
17
    data = { 'teste00' : [27], 'teste01' : [10], 'teste02' : [18], 'teste03' : [5], 'teste04' : [1], 'teste05' : [22] }
18
 
    CairoPlot.bar_plot ( 'bar_0_dictionary', data, 400, 300, border = 20, grid = True, rounded_corners = True )
 
18
    cairoplot.bar_plot ( 'bar_0_dictionary', data, 400, 300, border = 20, grid = True, rounded_corners = True )
19
19
 
20
20
    #Using default, rounded corners and 3D visualization
21
21
    data = [ [0, 3, 11], [8, 9, 21], [13, 10, 9], [2, 30, 8] ]
22
22
    colors = [ (1,0.2,0), (1,0.7,0), (1,1,0) ]
23
 
    CairoPlot.bar_plot ( 'bar_1_default', data, 400, 300, border = 20, grid = True, rounded_corners = False, colors = "yellow_orange_red" )
24
 
    CairoPlot.bar_plot ( 'bar_2_rounded', data, 400, 300, border = 20, grid = True, rounded_corners = True, colors = colors )
25
 
    CairoPlot.bar_plot ( 'bar_3_3D', data, 400, 300, border = 20, grid = True, three_dimension = True, colors = colors )
 
23
    cairoplot.bar_plot ( 'bar_1_default', data, 400, 300, border = 20, grid = True, rounded_corners = False, colors = "yellow_orange_red" )
 
24
    cairoplot.bar_plot ( 'bar_2_rounded', data, 400, 300, border = 20, grid = True, rounded_corners = True, colors = colors )
 
25
    cairoplot.bar_plot ( 'bar_3_3D', data, 400, 300, border = 20, grid = True, three_dimension = True, colors = colors )
26
26
 
27
27
    #Mixing groups and columns
28
28
    data = [ [1], [2], [3,4], [4], [5], [6], [7], [8], [9], [10] ]
29
 
    CairoPlot.bar_plot ( 'bar_4_group', data, 400, 300, border = 20, grid = True )
 
29
    cairoplot.bar_plot ( 'bar_4_group', data, 400, 300, border = 20, grid = True )
30
30
 
31
31
    #Using no labels, horizontal and vertical labels
32
32
    data = [[3,4], [4,8], [5,3], [9,1]]
33
33
    y_labels = [ "line1", "line2", "line3", "line4", "line5", "line6" ]
34
34
    x_labels = [ "group1", "group2", "group3", "group4" ]
35
 
    CairoPlot.bar_plot ( 'bar_5_no_labels', data, 600, 200, border = 20, grid = True )
36
 
    CairoPlot.bar_plot ( 'bar_6_x_labels', data, 600, 200, border = 20, grid = True, x_labels = x_labels )
37
 
    CairoPlot.bar_plot ( 'bar_7_y_labels', data, 600, 200, border = 20, grid = True, y_labels = y_labels )
38
 
    CairoPlot.bar_plot ( 'bar_8_hy_labels', data, 600, 200, border = 20, grid = True, x_labels = x_labels, y_labels = y_labels )
 
35
    cairoplot.bar_plot ( 'bar_5_no_labels', data, 600, 200, border = 20, grid = True )
 
36
    cairoplot.bar_plot ( 'bar_6_x_labels', data, 600, 200, border = 20, grid = True, x_labels = x_labels )
 
37
    cairoplot.bar_plot ( 'bar_7_y_labels', data, 600, 200, border = 20, grid = True, y_labels = y_labels )
 
38
    cairoplot.bar_plot ( 'bar_8_hy_labels', data, 600, 200, border = 20, grid = True, x_labels = x_labels, y_labels = y_labels )
39
39
 
40
40
if test_donut_plot :
41
41
    #Define a new backgrond
45
45
    
46
46
    data = {"john" : 700, "mary" : 100, "philip" : 100 , "suzy" : 50, "yman" : 50}
47
47
    #Default plot, gradient and shadow, different background
48
 
    CairoPlot.donut_plot( "donut_1_default", data, 600, 400, inner_radius = 0.3 )
49
 
    CairoPlot.donut_plot( "donut_2_gradient_shadow", data, 600, 400, gradient = True, shadow = True, inner_radius = 0.3 )
50
 
    CairoPlot.donut_plot( "donut_3_background", data, 600, 400, background = background, gradient = True, shadow = True, inner_radius = 0.3 )
 
48
    cairoplot.donut_plot( "donut_1_default", data, 600, 400, inner_radius = 0.3 )
 
49
    cairoplot.donut_plot( "donut_2_gradient_shadow", data, 600, 400, gradient = True, shadow = True, inner_radius = 0.3 )
 
50
    cairoplot.donut_plot( "donut_3_background", data, 600, 400, background = background, gradient = True, shadow = True, inner_radius = 0.3 )
51
51
 
52
52
if test_dot_line_plot:
53
53
    #Default plot
54
54
    data = [ 0, 1, 3.5, 8.5, 9, 0, 10, 10, 2, 1 ]
55
 
    CairoPlot.dot_line_plot( "dot_line_1_default", data, 400, 300, border = 50, axis = True, grid = True,
 
55
    cairoplot.dot_line_plot( "dot_line_1_default", data, 400, 300, border = 50, axis = True, grid = True,
56
56
                             x_title = "x axis", y_title = "y axis" )
57
57
 
58
58
    #Labels
59
59
    data = { "john" : [-5, -2, 0, 1, 3], "mary" : [0, 0, 3, 5, 2], "philip" : [-2, -3, -4, 2, 1] }
60
60
    x_labels = [ "jan/2008", "feb/2008", "mar/2008", "apr/2008", "may/2008" ]
61
61
    y_labels = [ "very low", "low", "medium", "high", "very high" ]
62
 
    CairoPlot.dot_line_plot( "dot_line_2_dictionary_labels", data, 400, 300, x_labels = x_labels, 
 
62
    cairoplot.dot_line_plot( "dot_line_2_dictionary_labels", data, 400, 300, x_labels = x_labels, 
63
63
                             y_labels = y_labels, axis = True, grid = True,
64
64
                             x_title = "x axis", y_title = "y axis", series_legend=True )
65
65
    
66
66
    #Series legend
67
67
    data = { "john" : [10, 10, 10, 10, 30], "mary" : [0, 0, 3, 5, 15], "philip" : [13, 32, 11, 25, 2] }
68
68
    x_labels = [ "jan/2008", "feb/2008", "mar/2008", "apr/2008", "may/2008" ]
69
 
    CairoPlot.dot_line_plot( 'dot_line_3_series_legend', data, 400, 300, x_labels = x_labels, 
 
69
    cairoplot.dot_line_plot( 'dot_line_3_series_legend', data, 400, 300, x_labels = x_labels, 
70
70
                             axis = True, grid = True, series_legend = True )
71
71
 
72
72
if test_function_plot :
73
73
    #Default Plot
74
74
    data = lambda x : x**2
75
 
    CairoPlot.function_plot( 'function_1_default', data, 400, 300, grid = True, x_bounds=(-10,10), step = 0.1 )
 
75
    cairoplot.function_plot( 'function_1_default', data, 400, 300, grid = True, x_bounds=(-10,10), step = 0.1 )
76
76
    
77
77
    #Discrete Plot
78
78
    data = lambda x : math.sin(0.1*x)*math.cos(x)
79
 
    CairoPlot.function_plot( 'function_2_discrete', data, 800, 300, discrete = True, dots = True, grid = True, x_bounds=(0,80), 
 
79
    cairoplot.function_plot( 'function_2_discrete', data, 800, 300, discrete = True, dots = True, grid = True, x_bounds=(0,80), 
80
80
                             x_title = "t (s)", y_title = "sin(0.1*x)*cos(x)")
81
81
 
82
82
    #Labels test
83
83
    data = lambda x : [1,2,3,4,5][x]
84
84
    x_labels = [ "4", "3", "2", "1", "0" ]
85
 
    CairoPlot.function_plot( 'function_3_labels', data, 400, 300, discrete = True, dots = True, grid = True, x_labels = x_labels, x_bounds=(0,4), step = 1 )
 
85
    cairoplot.function_plot( 'function_3_labels', data, 400, 300, discrete = True, dots = True, grid = True, x_labels = x_labels, x_bounds=(0,4), step = 1 )
86
86
    
87
87
    #Multiple functions
88
88
    data = [ lambda x : 1, lambda y : y**2, lambda z : -z**2 ]
89
89
    colors = [ (1.0, 0.0, 0.0 ), ( 0.0, 1.0, 0.0 ), ( 0.0, 0.0, 1.0 ) ]
90
 
    CairoPlot.function_plot( 'function_4_multi_functions', data, 400, 300, grid = True, series_colors = colors, step = 0.1 )
 
90
    cairoplot.function_plot( 'function_4_multi_functions', data, 400, 300, grid = True, series_colors = colors, step = 0.1 )
91
91
 
92
92
if test_gantt_chart :
93
93
    #Default Plot
95
95
    x_labels = [ 'teste01', 'teste02', 'teste03', 'teste04']
96
96
    y_labels = [ '0001', '0002', '0003', '0004', '0005', '0006', '0007', '0008', '0009', '0010' ]
97
97
    colors = [ (1.0, 0.0, 0.0), (1.0, 0.7, 0.0), (1.0, 1.0, 0.0), (0.0, 1.0, 0.0) ]
98
 
    CairoPlot.gantt_chart('gantt_1_default', pieces, 500, 350, x_labels, y_labels, colors)
 
98
    cairoplot.gantt_chart('gantt_1_default', pieces, 500, 350, x_labels, y_labels, colors)
99
99
 
100
100
 
101
101
if test_pie_plot :
106
106
 
107
107
    #Plot data
108
108
    data = {"orcs" : 100, "goblins" : 230, "elves" : 50 , "demons" : 43, "humans" : 332}
109
 
    CairoPlot.pie_plot( "pie_1_default", data, 600, 400 )
110
 
    CairoPlot.pie_plot( "pie_2_gradient_shadow", data, 600, 400, gradient = True, shadow = True )
111
 
    CairoPlot.pie_plot( "pie_3_background", data, 600, 400, background = background, gradient = True, shadow = True )
 
109
    cairoplot.pie_plot( "pie_1_default", data, 600, 400 )
 
110
    cairoplot.pie_plot( "pie_2_gradient_shadow", data, 600, 400, gradient = True, shadow = True )
 
111
    cairoplot.pie_plot( "pie_3_background", data, 600, 400, background = background, gradient = True, shadow = True )
112
112
 
113
113
if test_scatter_plot:
114
114
    #Default data
115
115
    data = [ (-2,10), (0,0), (0,15), (1,5), (2,0), (3,-10), (3,5) ]
116
 
    CairoPlot.scatter_plot ( 'cross_1_default', data = data, width = 500, height = 500, border = 20, axis = True, grid = True )
 
116
    cairoplot.scatter_plot ( 'cross_1_default', data = data, width = 500, height = 500, border = 20, axis = True, grid = True )
117
117
    
118
118
    data = [(-1, -16, 12), (-12, 17, 11), (-4, 6, 5), (4, -20, 12), (13, -3, 21), (7, 14, 20), (-11, -2, 18), (19, 7, 18), (-10, -19, 15),
119
119
            (-17, -2, 6), (-9, 4, 10), (14, 11, 16), (13, -11, 18), (20, 20, 16), (7, -8, 15), (-16, 17, 16), (16, 9, 9), (-3, -13, 25),
121
121
            (17, -15, 25), (-2, -8, 5), (5, 20, 20), (18, 20, 23), (-20, -16, 17), (-19, -2, 9), (-11, 19, 18), (17, 16, 12), (-5, -20, 15),
122
122
            (-20, -13, 10), (-3, 5, 20), (-1, 13, 17), (-11, -9, 11)]
123
123
    colors = [ (0,0,0,0.25), (1,0,0,0.75) ]
124
 
    CairoPlot.scatter_plot ( 'cross_2_variable_radius', data = data, width = 500, height = 500, border = 20, 
 
124
    cairoplot.scatter_plot ( 'cross_2_variable_radius', data = data, width = 500, height = 500, border = 20, 
125
125
                             axis = True, discrete = True, dots = 2, grid = True, 
126
126
                             x_title = "x axis", y_title = "y axis", circle_colors = colors )
127
127
    
134
134
    ery = [5*random.random() for x in t]
135
135
    data = {"exp" : [t,f], "cos" : [t,g], "sin" : [t,h]}
136
136
    series_colors = [ (1,0,0), (0,0,0), (0,0,1) ]
137
 
    CairoPlot.scatter_plot ( 'cross_r_exponential', data = data, errorx = [erx,erx], errory = [ery,ery], width = 800, height = 600, border = 20, 
 
137
    cairoplot.scatter_plot ( 'cross_r_exponential', data = data, errorx = [erx,erx], errory = [ery,ery], width = 800, height = 600, border = 20, 
138
138
                             axis = True, discrete = False, dots = 5, grid = True, 
139
139
                             x_title = "t", y_title = "f(t) g(t)", series_legend=True, series_colors = series_colors )
140
140
    
141
141
    
142
142
if test_themes :    
143
143
    data = [1,2,3,4,5,6,7,8,9,10,11,12,13,14]
144
 
    CairoPlot.bar_plot ( 'bar_color_themes', data, 400, 300, border = 20, grid = True, colors="rainbow" )
 
144
    cairoplot.bar_plot ( 'bar_color_themes', data, 400, 300, border = 20, grid = True, colors="rainbow" )
145
145
    
146
146
    data = [ lambda x : 1, lambda y : y**2, lambda z : -z**2 ]
147
 
    CairoPlot.function_plot( 'function_color_themes', data, 400, 300, grid = True, series_colors = ["red", "orange", "yellow"], step = 0.1 )
 
147
    cairoplot.function_plot( 'function_color_themes', data, 400, 300, grid = True, series_colors = ["red", "orange", "yellow"], step = 0.1 )
148
148
    
149
149
    #Scatter x DotLine
150
150
    t = [x*0.1 for x in range(0,40)]
155
155
    ery = [5*random.random() for x in t]
156
156
    data = {"exp" : [t,f], "cos" : [t,g], "sin" : [t,h]}
157
157
    series_colors = [ (1,0,0), (0,0,0) ]
158
 
    CairoPlot.scatter_plot ( 'scatter_color_themes', data = data, errorx = [erx,erx], errory = [ery,ery], width = 800, height = 600, border = 20, 
 
158
    cairoplot.scatter_plot ( 'scatter_color_themes', data = data, errorx = [erx,erx], errory = [ery,ery], width = 800, height = 600, border = 20, 
159
159
                             axis = True, discrete = False, dots = 5, grid = True, 
160
160
                             x_title = "t", y_title = "f(t) g(t)", series_legend=True, series_colors = ["red", "blue", "orange"])