clean up arrows pr

image/README.md

@@ -84,13 +84,14 @@ func main() {
     // write board SVG to file
     yellow := color.RGBA{255, 255, 0, 1}
     mark := image.MarkSquares(yellow, chess.D2, chess.D4)
-    if err := image.SVG(f, pos.Board(), mark); err != nil {
+    arrows := image.MarkArrows(image.Arrow(chess.D2, chess.D4))
+    if err := image.SVG(f, pos.Board(), mark, arrows); err != nil {
         log.Fatal(err)
     }
 }
 ```
 
 ### Example Program Result
 
-![rnbqkbnr/pppppppp/8/8/3P4/8/PPP1PPPP/RNBQKBNR b KQkq - 0 1](example.svg)
+![rnbqkbnr/pppppppp/8/8/3P4/8/PPP1PPPP/RNBQKBNR b KQkq - 0 1](examples/example.svg)
 

image/image.go

@@ -48,8 +48,8 @@ func (arrow arrow) WithColor(col color.Color) arrow {
 // MarkArrows is designed to be used as an optional argument
 // to the SVG function.  It marks an arrow between the given
 // squares with the an Arrow of the given color.
-func MarkArrows(arrows ...arrow) func(*encoder) {
-	return func(e *encoder) {
+func MarkArrows(arrows ...arrow) func(*Encoder) {
+	return func(e *Encoder) {
 		e.arrows = append(e.arrows, arrows...)
 	}
 }

image/image_test.go

@@ -5,8 +5,6 @@ import (
 	"crypto/md5"
 	"fmt"
 	"image/color"
-	"io"
-	"os"
 	"strings"
 	"testing"
 
@@ -38,16 +36,6 @@ func TestSVG(t *testing.T) {
 	if actualMD5 != expectedMD5 {
 		t.Errorf("expected actual md5 hash to be %s but got %s", expectedMD5, actualMD5)
 	}
-
-	// create actual svg file for visualization
-	f, err := os.Create("example.svg")
-	defer f.Close()
-	if err != nil {
-		t.Error(err)
-	}
-	if _, err := io.Copy(f, bytes.NewBufferString(actualSVG)); err != nil {
-		t.Error(err)
-	}
 }
 
 func TestSVGFromBlack(t *testing.T) {
@@ -71,16 +59,6 @@ func TestSVGFromBlack(t *testing.T) {
 	if actualMD5 != expectedMD5Black {
 		t.Errorf("expected actual md5 hash to be %s but got %s", expectedMD5Black, actualMD5)
 	}
-
-	// create actual svg file for visualization
-	f, err := os.Create("black_example.svg")
-	defer f.Close()
-	if err != nil {
-		t.Error(err)
-	}
-	if _, err := io.Copy(f, bytes.NewBufferString(actualSVG)); err != nil {
-		t.Error(err)
-	}
 }
 
 func TestSVGKnightsAndDiagonals(t *testing.T) {
@@ -125,14 +103,4 @@ func TestSVGKnightsAndDiagonals(t *testing.T) {
 	if actualMD5 != expectedMD5KnightsAndDiagonalArrows {
 		t.Errorf("expected actual md5 hash to be %s but got %s", expectedMD5KnightsAndDiagonalArrows, actualMD5)
 	}
-
-	// create actual svg file for visualization
-	f, err := os.Create("knight_arrows_example.svg")
-	defer f.Close()
-	if err != nil {
-		t.Error(err)
-	}
-	if _, err := io.Copy(f, bytes.NewBufferString(actualSVG)); err != nil {
-		t.Error(err)
-	}
 }