def test_outliner_mode(): # We can set up a pattern in one mode, p = Pen() p.set_mode(StrokeOutlineMode(sqrt3, 0.2 * sqrt3, 'blue', 'black')) p.move_to((0, 0)) p.turn_to(0) p.line_forward(5, end_slant=60) # Then continue it in another mode without caring what the first mode was. old_mode = p.mode p.set_mode(p.mode.outliner_mode()) p.turn_to(60) p.move_forward(1.0) p.turn_left(60) p.line_forward(2.0) p.turn_right(120) p.line_forward(2.0) p.turn_right(120) p.line_forward(2.0) p.turn_to(60) p.move_forward(3.0) p.turn_to(120) p.set_mode(old_mode) p.line_forward(5, start_slant=60) assert_svg_file( p, 3, 'test_outliner_mode.svg' )
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_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 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_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_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_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_custom_cap(): def circle_cap(pen, end): pen.arc_to(end) p = Pen() p.stroke_mode(2.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(5) p.last_segment().end_cap = circle_cap assert_path_data( p, 0, 'M0,-1 L0,1 L5,1 A 1,1 0 0 0 5,-1 L0,-1 z' ) p = Pen() p.stroke_mode(2.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(5) p.last_segment().start_cap = circle_cap assert_path_data( p, 0, 'M0,-1 A 1,1 0 1 0 0,1 L5,1 L5,-1 L0,-1 z' )
def test_start_slant_legal_joint(): # Create a joint that is only legal because of the start slant. p = Pen() p.outline_mode(1.0, 0.1) p.move_to((0, 0)) p.turn_to(0) p.line_forward(0.8, start_slant=-45) p.turn_left(90) p.outline_mode(2.0, 0.1) p.line_forward(5)
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_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_line_line_half_illegal_joint(): # The outside edge meets, but the inside is too short to meet. p = Pen() p.stroke_mode(1.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(2) p.turn_left(165) p.line_forward(2) assert_path_data( p, 2, ( 'M0.00,-0.50 L0.00,0.50 L5.80,0.50 L0.20,-1.00 ' 'L-0.06,-0.03 L1.87,0.48 L2.00,-0.50 L0.00,-0.50 z' ) ) p = Pen() p.stroke_mode(1.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(2) p.turn_right(165) p.line_forward(2) assert_path_data( p, 2, ( 'M0.00,-0.50 L0.00,0.50 L2.00,0.50 L1.87,-0.48 ' 'L-0.06,0.03 L0.20,1.00 L5.80,-0.50 L0.00,-0.50 z' ) )
def test_line_line_half_illegal_joint(): # The outside edge meets, but the inside is too short to meet. p = Pen() p.stroke_mode(1.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(2) p.turn_left(165) p.line_forward(2) assert_path_data(p, 2, ('M0.00,-0.50 L0.00,0.50 L5.80,0.50 L0.20,-1.00 ' 'L-0.06,-0.03 L1.87,0.48 L2.00,-0.50 L0.00,-0.50 z')) p = Pen() p.stroke_mode(1.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(2) p.turn_right(165) p.line_forward(2) assert_path_data(p, 2, ('M0.00,-0.50 L0.00,0.50 L2.00,0.50 L1.87,-0.48 ' 'L-0.06,0.03 L0.20,1.00 L5.80,-0.50 L0.00,-0.50 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_joint_loop_color(): p = Pen() p.move_to((0, 0)) p.turn_to(0) # Draw a square with one side a different color. It joins to the # beginning correctly. p.stroke_mode(2.0, color='black') p.line_forward(5) p.turn_left(90) p.line_forward(5) p.turn_left(90) p.line_forward(5) p.turn_left(90) p.stroke_mode(2.0, color='red') p.line_forward(5) assert_equal(len(p.paper.paths), 1) assert_path_data( p, 0, [ 'M1,-1 L-1,1 L6,1 L6,-6 L-1,-6 L1,-4 L4,-4 L4,-1 L1,-1 z', 'M1,-4 L-1,-6 L-1,1 L1,-1 L1,-4 z', ] )
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_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_multiple_strokes(): p = Pen() p.stroke_mode(1.0) p.turn_to(0) p.move_to((0, 0)) p.line_forward(3) p.move_to((0, 3)) p.line_forward(3) assert_path_data( p, 1, ( 'M0.0,-0.5 L0.0,0.5 L3.0,0.5 L3.0,-0.5 L0.0,-0.5 z ' 'M0.0,-3.5 L0.0,-2.5 L3.0,-2.5 L3.0,-3.5 L0.0,-3.5 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_mode_error(): p = Pen() # Don't set a mode. assert_raises( AttributeError, lambda: p.line_forward(1) )
def test_joint(): p = Pen() p.stroke_mode(1.0) p.move_to((-6, 0)) p.turn_to(0) p.line_forward(6) 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 L2.57,5.45 ' 'L3.43,4.95 L0.29,-0.50 L-6.00,-0.50 z' ), )
def test_break_stroke(): p = Pen() p.stroke_mode(2.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(3) p.break_stroke() p.line_forward(3) assert_path_data( p, 0, [ 'M0,-1 L0,1 L3,1 L3,-1 L0,-1 z', 'M3,-1 L3,1 L6,1 L6,-1 L3,-1 z', ] )
def test_straight_offwidth_no_joint(): p = Pen() p.move_to((0, 0)) p.turn_to(0) p.stroke_mode(2.0) p.line_forward(3) p.stroke_mode(1.0) p.line_forward(3) line1, line2 = p.last_path().segments assert line1.end_joint_illegal assert line2.start_joint_illegal assert_path_data(p, 1, ('M0.0,-1.0 L0.0,1.0 L3.0,1.0 L3.0,0.5 L6.0,0.5 ' 'L6.0,-0.5 L3.0,-0.5 L3.0,-1.0 L0.0,-1.0 z'))
def test_turn_back_no_joint(): # Make a line turn back on itself, and it doesn't join. p = Pen() p.stroke_mode(1.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(10) p.turn_right(180) p.line_forward(5) line1, line2 = p.last_path().segments assert line1.end_joint_illegal assert line2.start_joint_illegal assert_path_data(p, 1, ('M0.0,-0.5 L0.0,0.5 L10.0,0.5 L10.0,-0.5 ' 'L5.0,-0.5 L5.0,0.5 L10.0,0.5 L10.0,-0.5 L0.0,-0.5 z'))
def test_offwidth_joint(): p = Pen() p.stroke_mode(1.0) p.turn_to(0) p.move_forward(-3) p.line_forward(3) p.stroke_mode(0.5) p.turn_left(90) p.line_forward(3) assert_path_data( p, 2, ( 'M-3.00,-0.50 L-3.00,0.50 L0.25,0.50 L0.25,-3.00 ' 'L-0.25,-3.00 L-0.25,-0.50 L-3.00,-0.50 z' ), )
def test_change_mode(): # Change mode but don't change colors. It starts a new path. p = Pen() p.move_to((0, 0)) p.turn_to(0) p.stroke_mode(2.0, 'black') p.line_forward(5) p.fill_mode('black') p.line_forward(5) assert_equal( p.paper.svg_elements(0), [ '<path d="M0,-1 L0,1 L5,1 L5,-1 L0,-1 z" fill="#000000" />', '<path d="M5,0 L10,0" fill="#000000" />', ] )
def test_save_mode(): p = Pen() p.stroke_mode(2.0, 'red') old_mode = p.mode p.line_forward(5) p.fill_mode('blue') p.square(2) p.set_mode(old_mode) p.line_forward(5) assert_path_data( p, 0, [ 'M0,-1 L0,1 L5,1 L5,-1 L0,-1 z', 'M4,1 L6,1 L6,-1 L4,-1 L4,1 z', 'M5,-1 L5,1 L10,1 L10,-1 L5,-1 z', ] )
def test_arc_line_joint(): p = Pen() p.stroke_mode(1.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(3) p.turn_left(90) p.arc_left(180, 3) assert_path_data( p, 3, ( 'M0.000,-0.500 L0.000,0.500 L3.464,0.500 ' 'A 3.500,3.500 0 1 0 -3.500,0.000 L-2.500,0.000 ' 'A 2.500,2.500 0 0 1 2.449,-0.500 L0.000,-0.500 z' ), )
def test_log(): p = Pen() p.stroke_mode(1.0) p.move_to(Point(-6, 0)) p.turn_to(0) p.line_forward(6) p.turn_right(60) p.line_forward(6, end_slant=0) assert_equal( p.log(), [ 'stroke_mode(1.0)', 'move_to((-6, 0))', # Points are converted to tuples. 'turn_to(0)', 'line_forward(6)', 'turn_right(60)', 'line_forward(6, end_slant=0)', ] )
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_custom_cap(): def circle_cap(pen, end): pen.arc_to(end) p = Pen() p.stroke_mode(2.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(5) p.last_segment().end_cap = circle_cap assert_path_data(p, 0, 'M0,-1 L0,1 L5,1 A 1,1 0 0 0 5,-1 L0,-1 z') p = Pen() p.stroke_mode(2.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(5) p.last_segment().start_cap = circle_cap assert_path_data(p, 0, 'M0,-1 A 1,1 0 1 0 0,1 L5,1 L5,-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_line_thick(): p = Pen() p.stroke_mode(2.0) p.move_to((0, 0)) p.turn_to(0) p.line_forward(5) assert_path_data( p, 0, 'M0,-1 L0,1 L5,1 L5,-1 L0,-1 z' ) p = Pen() p.stroke_mode(1.0) p.turn_to(-45) p.line_forward(5) assert_path_data( p, 2, 'M0.35,-0.35 L-0.35,0.35 L3.18,3.89 L3.89,3.18 L0.35,-0.35 z' )
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' ) )