Clean up tests

summarize/readability_test.go

@@ -2,20 +2,13 @@ package summarize
 
 import (
 	"encoding/json"
-	"fmt"
 	"path/filepath"
 	"testing"
 
 	"github.com/jdkato/prose/internal/util"
 	"github.com/stretchr/testify/assert"
 )
 
-var testdata = filepath.Join("..", "testdata")
-
-func check(expected, observed float64) bool {
-	return fmt.Sprintf("%0.2f", expected) == fmt.Sprintf("%0.2f", observed)
-}
-
 func TestReadability(t *testing.T) {
 	tests := make([]testCase, 0)
 	cases := util.ReadDataFile(filepath.Join(testdata, "summarize.json"))

summarize/summarize_test.go

@@ -10,6 +10,12 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+var testdata = filepath.Join("..", "testdata")
+
+func check(expected, observed float64) bool {
+	return fmt.Sprintf("%0.2f", expected) == fmt.Sprintf("%0.2f", observed)
+}
+
 type testCase struct {
 	Text       string
 	Sentences  float64

summarize/syllables_test.go

@@ -8,7 +8,6 @@ import (
 	"testing"
 
 	"github.com/jdkato/prose/internal/util"
-	"github.com/jdkato/syllables"
 	"github.com/montanaflynn/stats"
 	"github.com/stretchr/testify/assert"
 )
@@ -87,7 +86,7 @@ func BenchmarkSyllablesIn(b *testing.B) {
 
 	for n := 0; n < b.N; n++ {
 		for word := range tests {
-			syllables.In(word)
+			Syllables(word)
 		}
 	}
 }

tokenize/regexp_test.go

@@ -1,34 +1,11 @@
 package tokenize
 
 import (
-	"encoding/json"
-	"path/filepath"
 	"testing"
 
-	"github.com/jdkato/prose/internal/util"
 	"github.com/stretchr/testify/assert"
 )
 
-func getWordData(file string) ([]string, [][]string) {
-	in := util.ReadDataFile(filepath.Join(testdata, "treebank_sents.json"))
-	out := util.ReadDataFile(filepath.Join(testdata, file))
-
-	input := []string{}
-	output := [][]string{}
-
-	util.CheckError(json.Unmarshal(in, &input))
-	util.CheckError(json.Unmarshal(out, &output))
-
-	return input, output
-}
-
-func getWordBenchData() []string {
-	in := util.ReadDataFile(filepath.Join(testdata, "treebank_sents.json"))
-	input := []string{}
-	util.CheckError(json.Unmarshal(in, &input))
-	return input
-}
-
 func TestWordPunctTokenizer(t *testing.T) {
 	input, output := getWordData("word_punct.json")
 	wordTokenizer := NewWordPunctTokenizer()

tokenize/tokenize_test.go

@@ -1,12 +1,37 @@
 package tokenize
 
 import (
+	"encoding/json"
 	"fmt"
+	"path/filepath"
 	"testing"
 
+	"github.com/jdkato/prose/internal/util"
 	"github.com/stretchr/testify/assert"
 )
 
+var testdata = filepath.Join("..", "testdata")
+
+func getWordData(file string) ([]string, [][]string) {
+	in := util.ReadDataFile(filepath.Join(testdata, "treebank_sents.json"))
+	out := util.ReadDataFile(filepath.Join(testdata, file))
+
+	input := []string{}
+	output := [][]string{}
+
+	util.CheckError(json.Unmarshal(in, &input))
+	util.CheckError(json.Unmarshal(out, &output))
+
+	return input, output
+}
+
+func getWordBenchData() []string {
+	in := util.ReadDataFile(filepath.Join(testdata, "treebank_sents.json"))
+	input := []string{}
+	util.CheckError(json.Unmarshal(in, &input))
+	return input
+}
+
 func ExampleNewWordBoundaryTokenizer() {
 	t := NewWordBoundaryTokenizer()
 	fmt.Println(t.Tokenize("They'll save and invest more."))

tokenize/treebank_test.go

@@ -1,14 +1,11 @@
 package tokenize
 
 import (
-	"path/filepath"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
-var testdata = filepath.Join("..", "testdata")
-
 func TestTreebankWordTokenizer(t *testing.T) {
 	input, output := getWordData("treebank_words.json")
 	word := NewTreebankWordTokenizer()

transform/title_test.go

@@ -9,8 +9,6 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-var testdata = filepath.Join("..", "testdata")
-
 type testCase struct {
 	Input  string
 	Expect string

transform/transform_test.go

@@ -1,6 +1,11 @@
 package transform
 
-import "fmt"
+import (
+	"fmt"
+	"path/filepath"
+)
+
+var testdata = filepath.Join("..", "testdata")
 
 func ExampleNewTitleConverter() {
 	tc := NewTitleConverter(APStyle)