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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
import unittest
import sys
import os
import warnings
ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
sys.path.insert(0, ROOT)
import pygooglechart as gc
class TestBase(unittest.TestCase):
def setUp(self):
# All tests require warnings to be raised
self.raise_warnings(True)
def raise_warnings(self, rw):
gc._reset_warnings()
if rw:
warnings.simplefilter('error')
else:
# Don't print out warnings if we're expecting them--so we can have
# nicer looking tests! :)
warnings.simplefilter('ignore')
class TestDataTypes(TestBase):
def test_simple_data(self):
s = gc.SimpleData([range(0, 62), [0, 1, 60, 61]])
self.assertEquals(repr(s),
'chd=s:ABCDEFGHIJKLMNOPQRSTUVWXYZ'
'abcdefghijklmnopqrstuvwxyz0123456789,AB89')
def test_text_data(self):
s = gc.TextData([[0, 1, 99.9]])
self.assertEquals(repr(s), 'chd=t:0.0,1.0,99.9')
def test_ext_data(self):
s = gc.ExtendedData([[0, 1, 4095]])
self.assertEquals(repr(s), 'chd=e:AAAB..')
class TestScaling(TestBase):
def test_simple_scale(self):
sv = gc.SimpleData.scale_value
self.raise_warnings(False) # We know some of these give warnings
self.assertEquals(sv(-10, [0, 1]), 0)
self.assertEquals(sv(0, [0, 1]), 0)
self.assertEquals(sv(.5, [0, 1]), 31)
self.assertEquals(sv(30, [0, 1]), 61)
self.assertEquals(sv(2222, [0, 10000]), 14)
# Test for warnings
self.raise_warnings(True)
self.assertRaises(UserWarning, sv, -10, [0, 1])
self.assertRaises(UserWarning, sv, 30, [0, 1])
def test_text_scale(self):
sv = gc.TextData.scale_value
self.raise_warnings(False)
self.assertEquals(sv(-10, [0, 1]), 0)
self.assertEquals(sv(0, [0, 1]), 0)
self.assertEquals(sv(.5, [0, 1]), 50)
self.assertEquals(sv(30, [0, 1]), 100)
self.assertEquals(sv(2222, [0, 10000]), 22.22)
self.raise_warnings(True)
sv(-10, [0, 1])
# self.assertRaises(UserWarning, sv, -10, [0, 1])
# self.assertEquals(UserWarning, sv, 30, [0, 1])
def test_ext_scale(self):
sv = gc.ExtendedData.scale_value
self.raise_warnings(False)
self.assertEquals(sv(-10, [0, 1]), 0)
self.assertEquals(sv(0, [0, 1]), 0)
self.assertEquals(sv(.5, [0, 1]), 2048)
self.assertEquals(sv(30, [0, 1]), 4095)
self.assertEquals(sv(2222, [0, 10000]), 910)
self.raise_warnings(True)
self.assertRaises(UserWarning, sv, -10, [0, 1])
self.assertRaises(UserWarning, sv, 30, [0, 1])
class TestGrammar(TestBase):
types = ('Venn', 'GroupedHorizontalBar', 'GoogleOMeter', 'Scatter',
'StackedVerticalBar', 'Map', 'StackedHorizontalBar', 'SimpleLine',
'SparkLine', 'GroupedVerticalBar', 'SplineRadar', 'XYLine', 'Radar')
def test_chart_types(self):
ret = gc.ChartGrammar.get_possible_chart_types()
diff = set(ret).symmetric_difference(set(TestGrammar.types))
self.assert_(not diff)
def test_google_chart(self):
g = {
'type': 'GoogleOMeter',
'w': 100,
'h': 100,
'auto_scale': True,
'x_range': [ 0, 10 ],
'data': [
[ 1, 5, 10 ]
],
}
grammar = gc.ChartGrammar()
chart = grammar.parse(g)
print chart.get_url()
# chart.download('meh.png')
if __name__ == "__main__":
unittest.main()
suite = unittest.TestSuite()
suite.addTest(TestScaling('test_ext_scale'))
unittest.TextTestRunner().run(suite)
|