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
|
import unittest, os, os.path, shutil
import _gdchartc
class uPyGDChart(unittest.TestCase):
SCRATCH = "scratch"
def setUp(self):
try:
os.mkdir(self.SCRATCH)
except OSError:
pass
def tearDown(self):
shutil.rmtree(self.SCRATCH)
def test_constants(self):
# Check a random constant...
self.failUnless(hasattr(_gdchartc, "GDC_LINE"))
# Check the single float constant
self.failUnless(hasattr(_gdchartc, "GDC_INTERP_VALUE"))
def test_options(self):
self.failUnless(_gdchartc.getOptions(_gdchartc.GRAPH))
self.failUnless(_gdchartc.getOptions(_gdchartc.PIE))
def test_out_graph_nonfile(self):
try:
_gdchartc.out_graph(1, 2, 1, 1, 1, ["one", "two"], 1, [1, 2], [1, 2])
except TypeError:
pass
else:
self.fail()
def test_out_graph_nonlist(self):
f = file(os.path.join(self.SCRATCH, "outgraph"), "w")
self.failUnlessRaises(TypeError, _gdchartc.out_graph, 1, 2, f, 1, 1, 2, 1, [1, 2], [1, 2])
self.failUnlessRaises(TypeError, _gdchartc.out_graph, 1, 2, f, 1, 1, [1, 2], 1, 2, [1, 2])
self.failUnlessRaises(TypeError, _gdchartc.out_graph, 1, 2, f, 1, 1, [1, 2], 1, [1, 2], 2)
_gdchartc.out_graph(100, 100, f, 1, 1, [1, 2], 1, [1, 2], [1, 2])
def test_out_graph_notfloats(self):
f = file(os.path.join(self.SCRATCH, "outgraph"), "w")
self.failUnlessRaises(TypeError, _gdchartc.out_graph, 1, 2, f, 1, 1, [1, 2], 1, ["one", "two"], [1, 2])
def test_out_graph_nulls(self):
f = file(os.path.join(self.SCRATCH, "outgraph"), "w")
_gdchartc.out_graph(100, 100, f, 1, 1, None, 1, [1, 2], [1, 2])
_gdchartc.out_graph(100, 100, f, 1, 1, None, 1, [1, 2], None)
_gdchartc.out_graph(100, 100, f, 1, 1, None, 1, [1, 2], [1, 2, None, 3, None])
self.failUnlessRaises(TypeError, _gdchartc.out_graph, 1, 2, f, 1, 1, None, 1, None, None)
def test_out_graph_succeed(self):
f = file(os.path.join(self.SCRATCH, "outgraph"), "w")
_gdchartc.out_graph(100, 100, f, 1, 1, ["one", "two"], 1, [1, 2], [1, 2])
class uPyGDChartOptions(unittest.TestCase):
def _testOption(self, setFunction, option, testvalue, chartType=_gdchartc.GRAPH):
opts = _gdchartc.getOptions(chartType)
val = opts[option][2]
setFunction(chartType, opts[option][0], testvalue)
opts = _gdchartc.getOptions(chartType)
self.failUnlessEqual(opts[option][2], testvalue)
# Now we restore the value...
setFunction(chartType, opts[option][0], val)
def test_setoptions(self):
self._testOption(_gdchartc.setOption, "grid", 99) # INT
self._testOption(_gdchartc.setOption, "xlabel_spacing", 99) # SHORT
self._testOption(_gdchartc.setOption, "perspective", 99, _gdchartc.PIE) # USHORT
self._testOption(_gdchartc.setOption, "threeD_angle", 20) # UCHAR
self._testOption(_gdchartc.setOption, "bg_transparent", 20) # BOOL
self._testOption(_gdchartc.setOption, "annotation_font_size", 20) # FONT
self._testOption(_gdchartc.setOption, "bar_width", 20) # PERCENT
self._testOption(_gdchartc.setOption, "other_threshold", 20, _gdchartc.PIE) # CHAR
self._testOption(_gdchartc.setOption, "bg_color", 20) # LONG
self._testOption(_gdchartc.setOption, "threeD_depth", 20) # FLOAT
self._testOption(_gdchartc.setOption, "bg_image", "foinklebar") # STRING
self._testOption(_gdchartc.setOption, "explode", range(500), _gdchartc.PIE) # INT_A
bools = [1, 1, 1, 1, 0, 1, 0, 1, 1, 0, 1, 0]
self._testOption(_gdchartc.setOption, "xlabel_ctl", bools) # BOOL_A
self._testOption(_gdchartc.setOption, "ext_color", range(500)) # COLORL_A
class Scat:
point = 0
val = 0
width = 0
color = 0
_type = 0
class uScatter(unittest.TestCase):
def test_scatter(self):
self.failUnlessRaises(TypeError, _gdchartc.setScatter, 1)
s = Scat()
s.point = "foo"
self.failUnlessRaises(TypeError, _gdchartc.setScatter, [s, Scat(), Scat(), Scat()])
_gdchartc.setScatter([Scat(), Scat(), Scat(), Scat()])
_gdchartc.setScatter(None)
class Anno:
point = 0
color = 0
note = "foo"
class uAnnotate(unittest.TestCase):
def test_anno(self):
_gdchartc.annotate(None)
a = Anno()
_gdchartc.annotate(a)
_gdchartc.annotate(None)
_gdchartc.annotate(None)
try:
_gdchartc.annotate(1)
except TypeError:
pass
_gdchartc.annotate(None)
|
