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 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_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_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_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_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_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_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" />', ] )