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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# sanity check
import pytest
def test_module_init():
import aggdraw
assert hasattr(aggdraw, 'VERSION')
assert isinstance(aggdraw.VERSION, str)
assert hasattr(aggdraw, '__version__')
assert isinstance(aggdraw.__version__, str)
def test_draw():
from aggdraw import Draw
from PIL import Image
with pytest.raises(AttributeError) as excinfo:
Draw("RGB")
assert "'str' object has no attribute 'mode'" in str(excinfo.value)
draw = Draw("RGB", (800, 600))
assert draw.mode == 'RGB'
assert draw.size == (800, 600)
draw = Draw("RGB", (800, 600), "white")
assert draw.mode == 'RGB'
assert draw.size == (800, 600)
im = Image.new("RGB", (600, 800))
draw = Draw(im)
assert draw.mode == 'RGB'
assert draw.size == (600, 800)
def test_flush():
from aggdraw import Draw
from PIL import Image
im = Image.new("RGB", (600, 800))
draw = Draw(im)
assert draw.flush().mode == 'RGB'
def test_pen():
from aggdraw import Pen
Pen("black")
Pen("black", 1)
Pen("black", 1.5)
Pen("black", 1, opacity=128)
Pen(0)
Pen((0,0,0))
Pen("rgb(0,0,0)")
Pen("gold")
def test_brush():
from aggdraw import Brush
Brush("black")
Brush("black", opacity=128)
Brush(0)
Brush((0, 0, 0))
Brush("rgb(0, 0, 0)")
Brush("gold")
def test_graphics():
from aggdraw import Draw, Pen, Brush
draw = Draw("RGB", (500, 500))
pen = Pen("black")
brush = Brush("black")
draw.line((50, 50, 100, 100), pen)
draw.rectangle((50, 150, 100, 200), pen)
draw.rectangle((50, 220, 100, 270), brush)
draw.rectangle((50, 290, 100, 340), brush, pen)
draw.rectangle((50, 360, 100, 410), pen, brush)
draw.ellipse((120, 150, 170, 200), pen)
draw.ellipse((120, 220, 170, 270), brush)
draw.ellipse((120, 290, 170, 340), brush, pen)
draw.ellipse((120, 360, 170, 410), pen, brush)
draw.polygon((190+25, 150, 190, 200, 190+50, 200), pen)
draw.polygon((190+25, 220, 190, 270, 190+50, 270), brush)
draw.polygon((190+25, 290, 190, 340, 190+50, 340), brush, pen)
draw.polygon((190+25, 360, 190, 410, 190+50, 410), pen, brush)
def test_graphics2():
"""See issue #14."""
from aggdraw import Draw, Symbol, Pen
from PIL import Image
import numpy as np
symbol = Symbol("M400 200 L400 400")
pen = Pen("red")
image = Image.fromarray(np.zeros((800, 600, 3), dtype=np.uint8), mode="RGB")
canvas = Draw(image)
canvas.symbol((0, 0), symbol, pen)
canvas.flush()
assert np.asarray(image).sum() == 50800
def test_graphics3():
"""See issue #22."""
from aggdraw import Draw, Pen
from PIL import Image
main = Image.new('RGB', (480, 1024), 'white')
d = Draw(main)
p = Pen((90,) * 3, 0.5)
def test_path():
from aggdraw import Draw, Path
Path()
Path([0, 0])
Path([0, 0, 0, 0])
p = Path()
p.moveto(0, 0)
p.lineto(1, 1)
assert p.coords() == [0.0, 0.0, 1.0, 1.0]
p.curveto(0, 0, 0, 0, 0, 0)
p.close()
p.coords()
assert p.coords() == [0.0, 0.0, 1.0, 1.0, 0.125, 0.125, 0.0, 0.0]
draw = Draw("RGB", (800, 600))
draw.line(p)
draw.polygon(p)
draw.symbol((0, 0), p)
def test_symbol():
from aggdraw import Symbol
Symbol("M0,0L0,0L0,0L0,0Z")
Symbol("M0,0L0,0,0,0,0,0Z", 10)
Symbol("M0,0C0,0,0,0,0,0Z")
Symbol("M0,0S0,0,0,0,0,0Z")
Symbol("m0,0l0,0l0,0l0,0z")
Symbol("m0,0l0,0,0,0,0,0z", 10)
Symbol("m0,0c0,0,0,0,0,0z")
Symbol("m0,0s0,0,0,0,0,0z")
def test_transform():
from aggdraw import Draw
draw = Draw("RGB", (500, 500))
draw.settransform()
draw.settransform((250, 250))
draw.settransform((1, 0, 250, 0, 1, 250))
draw.settransform((2.0, 0.5, 250, 0.5, 2.0, 250))
draw.settransform()
if __name__ == "__main__":
import sys
sys.exit(pytest.main(sys.argv))
|
