def test_straight_joint_headings():
# The math in calculating joint geometry can get numerically unstable
# very close to straight joints at various headings.
for heading_angle in range(0, 45):
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(heading_angle)
p.line_forward(10)
p.line_forward(10)
path = p.paper.paths[0]
path.render_path(2) # Doesn't crash.
# Check that the joint angle is 90 degrees from the heading.
assert_equal(len(p.paper.paths), 1)
segments = p.paper.paths[0].segments
assert_equal(len(segments), 2)
s0, s1 = segments
target_angle = (heading_angle + 90) % 180
joint_angle = math.degrees(vec.heading(vec.vfrom(s0.b_right, s0.b_left)))
assert_almost_equal(joint_angle % 180, target_angle)
joint_angle = math.degrees(vec.heading(vec.vfrom(s1.a_right, s1.a_left)))
assert_almost_equal(joint_angle % 180, target_angle)
def test_mode_error():
p = Pen()
# Don't set a mode.
assert_raises(
AttributeError,
lambda: p.line_forward(1)
)
def test_mode_error():
p = Pen()
# Don't set a mode.
assert_raises(
AttributeError,
lambda: p.line_forward(1)
)
def draw(offset):
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(0.5 + offset, end_slant=45)
return p
def test_color_formats():
for color, output in [
(
(1.0, 0.0, 0.0),
'#ff0000',
),
(
Color.from_html('red'),
'#ff0000',
),
(
'green',
'#008000',
),
(
'#123456',
'#123456',
),
]:
p = Pen()
p.stroke_mode(2.0, color)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(5)
assert_equal(
p.paper.svg_elements(0)[0],
'<path d="M0,-1 L0,1 L5,1 L5,-1 L0,-1 z" fill="{}" />'.format(output)
)
def test_circle_degenerate():
p = Pen()
p.stroke_mode(2.0)
p.circle(1)
assert_equal(
p.paper.svg_elements(0),
['<path d="M2,0 A 2,2 0 0 0 -2,0 A 2,2 0 0 0 2,0 z" fill="#000000" />']
)
def test_circle():
p = Pen()
p.fill_mode()
p.circle(1)
assert_equal(
p.paper.svg_elements(0),
['<path d="M1,0 A 1,1 0 0 0 -1,0 A 1,1 0 0 0 1,0 z" fill="#000000" />']
)
def test_color_formats():
for color, output in [
(
(1.0, 0.0, 0.0),
'#ff0000',
),
(
Color.NewFromHtml('red'),
'#ff0000',
),
(
'green',
'#008000',
),
(
'#123456',
'#123456',
),
]:
p = Pen()
p.stroke_mode(2.0, color)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(5)
assert_equal(
p.paper.svg_elements(0)[0],
'<path d="M0,-1 L0,1 L5,1 L5,-1 L0,-1 z" fill="{}" />'.format(output)
)
def test_straight_joint_headings():
# The math in calculating joint geometry can get numerically unstable
# very close to straight joints at various headings.
for heading_angle in range(0, 45):
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(heading_angle)
p.line_forward(10)
p.line_forward(10)
path = p.paper.paths[0]
path.render_path(2) # Doesn't crash.
# Check that the joint angle is 90 degrees from the heading.
assert_equal(len(p.paper.paths), 1)
segments = p.paper.paths[0].segments
assert_equal(len(segments), 2)
s0, s1 = segments
target_angle = (heading_angle + 90) % 180
joint_angle = math.degrees(vec.heading(vec.vfrom(s0.b_right, s0.b_left)))
assert_almost_equal(joint_angle % 180, target_angle)
joint_angle = math.degrees(vec.heading(vec.vfrom(s1.a_right, s1.a_left)))
assert_almost_equal(joint_angle % 180, target_angle)
def draw(offset):
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(0.5 + offset, end_slant=45)
return p
def test_arc_error():
# Don't allow drawing an arc without a center or radius.
p = Pen()
assert_raises(
TypeError,
lambda: p.arc_left(90)
)
assert_raises(
TypeError,
lambda: p.arc_right(90)
)
def test_arc_error():
# Don't allow drawing an arc without a center or radius.
p = Pen()
assert_raises(
TypeError,
lambda: p.arc_left(90)
)
assert_raises(
TypeError,
lambda: p.arc_right(90)
)
def test_line():
p = Pen()
p.fill_mode()
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(5)
assert_path_data(
p, 0,
'M0,0 L5,0'
)
def test_keep_mode_colors():
# Setting a mode with no color specified usually gives a default color.
p = Pen()
p.stroke_mode(1.0)
assert_equal(p.mode.color, None)
# You can change mode width only, without affecting color.
p = Pen()
p.stroke_mode(1.0, 'red')
p.stroke_mode(0.5)
assert_equal(p.mode.width, 0.5)
assert_equal(p.mode.color, 'red')
def test_arc_sweep_bug():
p = Pen()
p.stroke_mode(2.0)
p.move_to((3, 0))
p.turn_to(90)
p.arc_left(270, 3)
assert_path_data(
p, 0,
'M2,0 L4,0 A 4,4 0 1 0 0,4 L0,2 A 2,2 0 1 1 2,0 z'
)
def test_arc_normalize():
# Arc angles larger than 360 behave correctly.
p = Pen()
p.fill_mode()
p.move_to((-5, 0))
p.turn_to(0)
p.arc_left(360 + 90, radius=5)
assert_path_data(
p, 0,
'M-5,0 A 5,5 0 0 0 0,-5'
)
def test_straight_joint():
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(3)
p.line_forward(3)
assert_path_data(
p, 0,
'M0,-1 L0,1 L3,1 L6,1 L6,-1 L3,-1 L0,-1 z'
)
def test_zero_length_side():
# It is possible and legal to create a segment that just barely goes to
# zero on one side.
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(1.0, end_slant=45)
assert_path_data(
p, 0,
'M0,-1 L0,1 L2,-1 L0,-1 z',
)
def test_arc_pie_slice():
# Draw a "pie slice" arc that is wide enough to reach all the way to the
# arc center.
p = Pen()
p.stroke_mode(1.0)
p.move_to((0.5, 0))
p.turn_to(90)
p.arc_left(90, 0.5)
assert_path_data(
p, 0,
'M0,0 L1,0 A 1,1 0 0 0 0,-1 L0,0 z'
)
def test_outline():
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.outline_mode(1.0, 0.2)
p.line_forward(3)
assert_path_data(
p, 1,
(
'M-0.1,-0.6 L-0.1,0.6 L3.1,0.6 L3.1,-0.6 L-0.1,-0.6 z '
'M0.1,-0.4 L2.9,-0.4 L2.9,0.4 L0.1,0.4 L0.1,-0.4 z'
)
)
def test_arc_angle():
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(0)
p.arc_left(90, radius=5, start_slant=45, end_slant=45)
assert_path_data(
p, 2,
(
'M0.53,-0.53 L-0.48,0.48 A 5.50,5.50 0 0 0 5.48,-5.48 '
'L4.47,-4.47 A 4.50,4.50 0 0 1 0.53,-0.53 z'
),
)
def test_stroke_fill_mode():
p = Pen()
p.set_mode(StrokeFillMode(0.2, 'black', 'red'))
p.move_to((0, 0))
p.turn_to(0)
p.arc_left(180, 5)
assert_equal(
p.paper.svg_elements(1)[0],
(
'<path d="M0.0,0.0 A 5.0,5.0 0 0 0 0.0,-10.0" fill="#ff0000" />'
'<path d="M0.0,-0.1 L0.0,0.1 A 5.1,5.1 0 0 0 0.0,-10.1 '
'L0.0,-9.9 A 4.9,4.9 0 0 1 0.0,-0.1 z" fill="#000000" />'
)
)
def test_joint_loop_multiple():
p = Pen()
p.stroke_mode(0.2)
def square():
p.turn_to(180)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.move_to((0, 0))
square()
p.move_to((2, 0))
square()
assert_path_data(
p, 1,
(
'M0.1,-0.1 L-1.1,-0.1 L-1.1,1.1 L0.1,1.1 L0.1,-0.1 z '
'M-0.1,0.1 L-0.1,0.9 L-0.9,0.9 L-0.9,0.1 L-0.1,0.1 z '
'M2.1,-0.1 L0.9,-0.1 L0.9,1.1 L2.1,1.1 L2.1,-0.1 z '
'M1.9,0.1 L1.9,0.9 L1.1,0.9 L1.1,0.1 L1.9,0.1 z'
)
)
def test_line():
p = Pen()
p.fill_mode()
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(5)
assert_path_data(
p, 0,
'M0,0 L5,0'
)
def test_degenerate_arc():
p = Pen()
p.stroke_mode(2.0)
p.move_to((-5, 0))
p.turn_to(0)
p.arc_to(
(5, 0),
center=(0, -200),
start_slant=-5,
end_slant=5,
)
seg = p.last_segment()
assert seg.start_joint_illegal
assert seg.end_joint_illegal
def test_line_segments():
p = Pen()
p.fill_mode()
p.move_to((0, 0))
p.turn_to(45)
p.line_forward(2.0)
assert_points_equal(p.position, (sqrt2, sqrt2))
assert_equal(len(p.paper.paths), 1)
segments = p.last_path().segments
for actual, target in zip(segments, [
((0, 0), (sqrt2, sqrt2)),
]):
assert_segments_equal(actual, target)
def test_arc_normalize():
# Arc angles larger than 360 behave correctly.
p = Pen()
p.fill_mode()
p.move_to((-5, 0))
p.turn_to(0)
p.arc_left(360 + 90, radius=5)
assert_path_data(
p, 0,
'M-5,0 A 5,5 0 0 0 0,-5'
)
def test_arc_sweep_bug():
p = Pen()
p.stroke_mode(2.0)
p.move_to((3, 0))
p.turn_to(90)
p.arc_left(270, 3)
assert_path_data(
p, 0,
'M2,0 L4,0 A 4,4 0 1 0 0,4 L0,2 A 2,2 0 1 1 2,0 z'
)
def test_straight_joint():
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(3)
p.line_forward(3)
assert_path_data(
p, 0,
'M0,-1 L0,1 L3,1 L6,1 L6,-1 L3,-1 L0,-1 z'
)
def test_stroke_outline_mode():
p = Pen()
p.set_mode(StrokeOutlineMode(1.0, 0.2, 'red', 'black'))
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(5)
assert_equal(
p.paper.svg_elements(1)[0],
(
'<path d="M0.0,-0.5 L0.0,0.5 L5.0,0.5 L5.0,-0.5 L0.0,-0.5 z" '
'fill="#ff0000" />'
'<path d="M-0.1,-0.6 L-0.1,0.6 L5.1,0.6 L5.1,-0.6 L-0.1,-0.6 z '
'M0.1,-0.4 L4.9,-0.4 L4.9,0.4 L0.1,0.4 L0.1,-0.4 z" '
'fill="#000000" />'
)
)
def test_zero_length_side():
# It is possible and legal to create a segment that just barely goes to
# zero on one side.
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(1.0, end_slant=45)
assert_path_data(
p, 0,
'M0,-1 L0,1 L2,-1 L0,-1 z',
)
def test_arc_pie_slice():
# Draw a "pie slice" arc that is wide enough to reach all the way to the
# arc center.
p = Pen()
p.stroke_mode(1.0)
p.move_to((0.5, 0))
p.turn_to(90)
p.arc_left(90, 0.5)
assert_path_data(
p, 0,
'M0,0 L1,0 A 1,1 0 0 0 0,-1 L0,0 z'
)
def test_stroke_fill_mode():
p = Pen()
p.set_mode(StrokeFillMode(0.2, 'black', 'red'))
p.move_to((0, 0))
p.turn_to(0)
p.arc_left(180, 5)
assert_equal(
p.paper.svg_elements(1)[0],
(
'<path d="M0.0,0.0 A 5.0,5.0 0 0 0 0.0,-10.0" fill="#ff0000" />'
'<path d="M0.0,-0.1 L0.0,0.1 A 5.1,5.1 0 0 0 0.0,-10.1 '
'L0.0,-9.9 A 4.9,4.9 0 0 1 0.0,-0.1 z" fill="#000000" />'
)
)
def test_outline():
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.outline_mode(1.0, 0.2)
p.line_forward(3)
assert_path_data(
p, 1,
(
'M-0.1,-0.6 L-0.1,0.6 L3.1,0.6 L3.1,-0.6 L-0.1,-0.6 z '
'M0.1,-0.4 L2.9,-0.4 L2.9,0.4 L0.1,0.4 L0.1,-0.4 z'
)
)
def test_arc_angle():
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(0)
p.arc_left(90, radius=5, start_slant=45, end_slant=45)
assert_path_data(
p, 2,
(
'M0.53,-0.53 L-0.48,0.48 A 5.50,5.50 0 0 0 5.48,-5.48 '
'L4.47,-4.47 A 4.50,4.50 0 0 1 0.53,-0.53 z'
),
)
def test_joint_loop_multiple():
p = Pen()
p.stroke_mode(0.2)
def square():
p.turn_to(180)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.move_to((0, 0))
square()
p.move_to((2, 0))
square()
assert_path_data(
p, 1,
(
'M0.1,-0.1 L-1.1,-0.1 L-1.1,1.1 L0.1,1.1 L0.1,-0.1 z '
'M-0.1,0.1 L-0.1,0.9 L-0.9,0.9 L-0.9,0.1 L-0.1,0.1 z '
'M2.1,-0.1 L0.9,-0.1 L0.9,1.1 L2.1,1.1 L2.1,-0.1 z '
'M1.9,0.1 L1.9,0.9 L1.1,0.9 L1.1,0.1 L1.9,0.1 z'
)
)
def test_keep_mode_colors():
# Setting a mode with no color specified usually gives a default color.
p = Pen()
p.stroke_mode(1.0)
assert_equal(p.mode.color, None)
# You can change mode width only, without affecting color.
p = Pen()
p.stroke_mode(1.0, 'red')
p.stroke_mode(0.5)
assert_equal(p.mode.width, 0.5)
assert_equal(p.mode.color, 'red')
def test_stroke_outline_mode():
p = Pen()
p.set_mode(StrokeOutlineMode(1.0, 0.2, 'red', 'black'))
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(5)
assert_equal(
p.paper.svg_elements(1)[0],
(
'<path d="M0.0,-0.5 L0.0,0.5 L5.0,0.5 L5.0,-0.5 L0.0,-0.5 z" '
'fill="#ff0000" />'
'<path d="M-0.1,-0.6 L-0.1,0.6 L5.1,0.6 L5.1,-0.6 L-0.1,-0.6 z '
'M0.1,-0.4 L4.9,-0.4 L4.9,0.4 L0.1,0.4 L0.1,-0.4 z" '
'fill="#000000" />'
)
)
def test_circle_degenerate():
p = Pen()
p.stroke_mode(2.0)
p.circle(1)
assert_equal(
p.paper.svg_elements(0),
['<path d="M2,0 A 2,2 0 0 0 -2,0 A 2,2 0 0 0 2,0 z" fill="#000000" />']
)
def test_circle():
p = Pen()
p.fill_mode()
p.circle(1)
assert_equal(
p.paper.svg_elements(0),
['<path d="M1,0 A 1,1 0 0 0 -1,0 A 1,1 0 0 0 1,0 z" fill="#000000" />']
)
def test_long_line_thick():
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
for _ in range(2):
p.line_forward(5)
p.turn_right(90)
p.line_forward(5)
p.turn_left(90)
assert_path_data(
p, 0,
'M0,-1 L0,1 L4,1 L4,6 L9,6 L9,10 L11,10 L11,4 L6,4 L6,-1 L0,-1 z'
)
def test_arc_joint_continue():
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
p.arc_left(90, 5)
p.arc_left(90, 5)
p.move_to((0, 0))
p.turn_to(0)
p.arc_right(90, 5)
p.arc_right(90, 5)
assert_path_data(
p, 0,
(
'M0,-1 L0,1 A 6,6 0 0 0 6,-5 A 6,6 0 0 0 0,-11 '
'L0,-9 A 4,4 0 0 1 4,-5 A 4,4 0 0 1 0,-1 z '
'M0,-1 L0,1 A 4,4 0 0 1 4,5 A 4,4 0 0 1 0,9 '
'L0,11 A 6,6 0 0 0 6,5 A 6,6 0 0 0 0,-1 z'
),
)
def test_color_path():
# Changing colors starts a new path.
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.stroke_mode(1.0, (1.0, 0.0, 0.0))
p.line_forward(1)
p.stroke_mode(1.0, (0.0, 1.0, 0.0))
p.line_forward(1)
p.stroke_mode(1.0, (0.0, 0.0, 1.0))
p.line_forward(1)
assert_equal(
p.paper.svg_elements(1),
[
(
'<path d="M0.0,-0.5 L0.0,0.5 L1.0,0.5 L1.0,-0.5 L0.0,-0.5 z" '
'fill="#ff0000" />'
'<path d="M1.0,-0.5 L1.0,0.5 L2.0,0.5 L2.0,-0.5 L1.0,-0.5 z" '
'fill="#00ff00" />'
'<path d="M2.0,-0.5 L2.0,0.5 L3.0,0.5 L3.0,-0.5 L2.0,-0.5 z" '
'fill="#0000ff" />'
),
]
)
def test_circle_line_overlap():
# Draw a circle that is above one line but below the other line.
p = Pen()
p.stroke_mode(1.0, color=(1.0, 0.0, 0.0))
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(4)
p.fill_mode(color=(0.0, 1.0, 0.0))
p.move_to((2, 2))
p.circle(2)
p.stroke_mode(1.0, color=(0.0, 0.0, 1.0))
p.move_to((0, 4))
p.turn_to(0)
p.line_forward(4)
assert_equal(
p.paper.svg_elements(1),
[
(
'<path d="M0.0,-0.5 L0.0,0.5 L4.0,0.5 L4.0,-0.5 L0.0,-0.5 z" '
'fill="#ff0000" />'
),
(
'<path d="M4.0,-2.0 A 2.0,2.0 0 0 0 0.0,-2.0 '
'A 2.0,2.0 0 0 0 4.0,-2.0 z" fill="#00ff00" />'
),
(
'<path d="M0.0,-4.5 L0.0,-3.5 L4.0,-3.5 L4.0,-4.5 L0.0,-4.5 z" '
'fill="#0000ff" />'
),
]
)
def test_color_joint():
p = Pen()
p.stroke_mode(1.0, 'red')
p.move_to((-6, 0))
p.turn_to(0)
p.line_forward(6)
p.stroke_mode(1.0, 'green')
p.turn_right(60)
p.line_forward(6)
assert_path_data(
p, 2,
[
'M-6.00,-0.50 L-6.00,0.50 L-0.29,0.50 L0.29,-0.50 L-6.00,-0.50 z',
'M0.29,-0.50 L-0.29,0.50 L2.57,5.45 L3.43,4.95 L0.29,-0.50 z',
]
)
def test_arc_arc_joint_off_radius():
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(0)
p.arc_left(180, 1)
p.arc_left(90, 2)
assert_path_data(
p, 1,
(
'M0.0,-0.5 L0.0,0.5 '
'A 1.5,1.5 0 0 0 0.0,-2.5 '
'A 2.5,2.5 0 0 0 -2.5,0.0 '
'L-1.5,0.0 '
'A 1.5,1.5 0 0 1 0.0,-1.5 '
'A 0.5,0.5 0 0 1 0.0,-0.5 z'
)
)
def test_repr():
p = Pen()
p.fill_mode()
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(1)
p.arc_left(90, 1)
path = p.paper.paths[0]
line, arc = path.segments
assert_equal(
repr(line),
'LineSegment(a=Point(x=0, y=0), b=Point(x=1.0, y=0.0))'
)
assert_equal(
repr(arc),
(
'ArcSegment(a=Point(x=1.0, y=0.0), b=Point(x=2.0, y=0.9999999999999999), '
'center=Point(x=1.0, y=1.0), radius=1, start_heading=0, end_heading=90)'
)
)
def test_various_joins():
p = Pen()
p.stroke_mode(0.5)
p.move_to((-2, 0))
p.turn_to(0)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.turn_right(90)
p.arc_right(90, 1)
p.arc_left(90, 1)
p.turn_left(90)
p.line_forward(1)
p.paper.override_bounds(-3, -3, 3, 3)
assert_svg_file(
p, 2,
'test_various_joins.svg',
)
def test_offwidth_arc_joins():
# Join arcs and lines of different widths.
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.stroke_mode(0.8)
p.line_forward(5)
p.turn_left(45)
p.stroke_mode(3.0)
p.arc_left(90, 5)
p.turn_to(-180)
p.line_forward(5)
p.turn_left(45)
p.stroke_mode(0.8)
p.arc_left(45, 5)
p.turn_right(90)
p.stroke_mode(3.0)
p.arc_right(90, 4)
assert_svg_file(
p, 3,
'test_offwidth_arc_joins.svg'
)
def test_circle_line_overlap():
# Draw a circle that is above one line but below the other line.
p = Pen()
p.stroke_mode(1.0, color=(1.0, 0.0, 0.0))
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(4)
p.fill_mode(color=(0.0, 1.0, 0.0))
p.move_to((2, 2))
p.circle(2)
p.stroke_mode(1.0, color=(0.0, 0.0, 1.0))
p.move_to((0, 4))
p.turn_to(0)
p.line_forward(4)
assert_equal(
p.paper.svg_elements(1),
[
(
'<path d="M0.0,-0.5 L0.0,0.5 L4.0,0.5 L4.0,-0.5 L0.0,-0.5 z" '
'fill="#ff0000" />'
),
(
'<path d="M4.0,-2.0 A 2.0,2.0 0 0 0 0.0,-2.0 '
'A 2.0,2.0 0 0 0 4.0,-2.0 z" fill="#00ff00" />'
),
(
'<path d="M0.0,-4.5 L0.0,-3.5 L4.0,-3.5 L4.0,-4.5 L0.0,-4.5 z" '
'fill="#0000ff" />'
),
]
)
def test_arc_line_joint_bug():
# When using arc_to, sometimes the b_left and b_right would get
# reversed.
p = Pen()
p.stroke_mode(1.0)
p.move_to((0, 0))
p.turn_to(90)
p.arc_to((5, 5))
p.turn_to(-90)
p.line_forward(5)
assert_path_data(
p, 3,
(
'M-0.500,0.000 L0.500,0.000 '
'A 4.500,4.500 0 0 1 4.500,-4.472 '
'L4.500,0.000 L5.500,0.000 L5.500,-5.477 '
'A 5.500,5.500 0 0 0 -0.500,0.000 z'
)
)
def test_color_path():
# Changing colors starts a new path.
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.stroke_mode(1.0, (1.0, 0.0, 0.0))
p.line_forward(1)
p.stroke_mode(1.0, (0.0, 1.0, 0.0))
p.line_forward(1)
p.stroke_mode(1.0, (0.0, 0.0, 1.0))
p.line_forward(1)
assert_equal(
p.paper.svg_elements(1),
[
(
'<path d="M0.0,-0.5 L0.0,0.5 L1.0,0.5 L1.0,-0.5 L0.0,-0.5 z" '
'fill="#ff0000" />'
'<path d="M1.0,-0.5 L1.0,0.5 L2.0,0.5 L2.0,-0.5 L1.0,-0.5 z" '
'fill="#00ff00" />'
'<path d="M2.0,-0.5 L2.0,0.5 L3.0,0.5 L3.0,-0.5 L2.0,-0.5 z" '
'fill="#0000ff" />'
),
]
)
def test_circle_color():
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.fill_mode((1.0, 0.0, 0.0))
p.circle(1)
p.move_forward(2)
p.fill_mode((0.0, 1.0, 0.0))
p.circle(1)
p.move_forward(2)
p.fill_mode((0.0, 0.0, 1.0))
p.circle(1)
assert_equal(
p.paper.svg_elements(0),
[
'<path d="M1,0 A 1,1 0 0 0 -1,0 A 1,1 0 0 0 1,0 z" fill="#ff0000" />',
'<path d="M3,0 A 1,1 0 0 0 1,0 A 1,1 0 0 0 3,0 z" fill="#00ff00" />',
'<path d="M5,0 A 1,1 0 0 0 3,0 A 1,1 0 0 0 5,0 z" fill="#0000ff" />',
]
)
def test_color_joint():
p = Pen()
p.stroke_mode(1.0, 'red')
p.move_to((-6, 0))
p.turn_to(0)
p.line_forward(6)
p.stroke_mode(1.0, 'green')
p.turn_right(60)
p.line_forward(6)
assert_path_data(
p, 2,
[
'M-6.00,-0.50 L-6.00,0.50 L-0.29,0.50 L0.29,-0.50 L-6.00,-0.50 z',
'M0.29,-0.50 L-0.29,0.50 L2.57,5.45 L3.43,4.95 L0.29,-0.50 z',
]
)
def test_long_line_thick():
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
for _ in range(2):
p.line_forward(5)
p.turn_right(90)
p.line_forward(5)
p.turn_left(90)
assert_path_data(
p, 0,
'M0,-1 L0,1 L4,1 L4,6 L9,6 L9,10 L11,10 L11,4 L6,4 L6,-1 L0,-1 z'
)
def test_repr():
p = Pen()
p.fill_mode()
p.move_to((0, 0))
p.turn_to(0)
p.line_forward(1)
p.arc_left(90, 1)
path = p.paper.paths[0]
line, arc = path.segments
assert_equal(
repr(line),
'LineSegment(a=Point(x=0, y=0), b=Point(x=1.0, y=0.0))'
)
assert_equal(
repr(arc),
(
'ArcSegment(a=Point(x=1.0, y=0.0), b=Point(x=2.0, y=0.9999999999999999), '
'center=Point(x=1.0, y=1.0), radius=1, start_heading=0, end_heading=90)'
)
)
def test_offwidth_arc_joins():
# Join arcs and lines of different widths.
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.stroke_mode(0.8)
p.line_forward(5)
p.turn_left(45)
p.stroke_mode(3.0)
p.arc_left(90, 5)
p.turn_to(-180)
p.line_forward(5)
p.turn_left(45)
p.stroke_mode(0.8)
p.arc_left(45, 5)
p.turn_right(90)
p.stroke_mode(3.0)
p.arc_right(90, 4)
assert_svg_file(
p, 3,
'test_offwidth_arc_joins.svg'
)
def test_various_joins():
p = Pen()
p.stroke_mode(0.5)
p.move_to((-2, 0))
p.turn_to(0)
p.line_forward(1)
p.turn_left(90)
p.line_forward(1)
p.turn_right(90)
p.arc_right(90, 1)
p.arc_left(90, 1)
p.turn_left(90)
p.line_forward(1)
p.paper.override_bounds(-3, -3, 3, 3)
assert_svg_file(
p, 2,
'test_various_joins.svg',
)
def test_arc_joint_continue():
p = Pen()
p.stroke_mode(2.0)
p.move_to((0, 0))
p.turn_to(0)
p.arc_left(90, 5)
p.arc_left(90, 5)
p.move_to((0, 0))
p.turn_to(0)
p.arc_right(90, 5)
p.arc_right(90, 5)
assert_path_data(
p, 0,
(
'M0,-1 L0,1 A 6,6 0 0 0 6,-5 A 6,6 0 0 0 0,-11 '
'L0,-9 A 4,4 0 0 1 4,-5 A 4,4 0 0 1 0,-1 z '
'M0,-1 L0,1 A 4,4 0 0 1 4,5 A 4,4 0 0 1 0,9 '
'L0,11 A 6,6 0 0 0 6,5 A 6,6 0 0 0 0,-1 z'
),
)
def test_circle_color():
p = Pen()
p.move_to((0, 0))
p.turn_to(0)
p.fill_mode((1.0, 0.0, 0.0))
p.circle(1)
p.move_forward(2)
p.fill_mode((0.0, 1.0, 0.0))
p.circle(1)
p.move_forward(2)
p.fill_mode((0.0, 0.0, 1.0))
p.circle(1)
assert_equal(
p.paper.svg_elements(0),
[
'<path d="M1,0 A 1,1 0 0 0 -1,0 A 1,1 0 0 0 1,0 z" fill="#ff0000" />',
'<path d="M3,0 A 1,1 0 0 0 1,0 A 1,1 0 0 0 3,0 z" fill="#00ff00" />',
'<path d="M5,0 A 1,1 0 0 0 3,0 A 1,1 0 0 0 5,0 z" fill="#0000ff" />',
]
)
