63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
|
package text
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var text = "The quick brown fox jumps over the lazy dog."
|
||
|
|
|
||
|
|
func TestWrap(t *testing.T) {
|
||
|
|
exp := [][]string{
|
||
|
|
{"The", "quick", "brown", "fox"},
|
||
|
|
{"jumps", "over", "the", "lazy", "dog."},
|
||
|
|
}
|
||
|
|
words := bytes.Split([]byte(text), sp)
|
||
|
|
got := WrapWords(words, 1, 24, defaultPenalty)
|
||
|
|
if len(exp) != len(got) {
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
for i := range exp {
|
||
|
|
if len(exp[i]) != len(got[i]) {
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
for j := range exp[i] {
|
||
|
|
if exp[i][j] != string(got[i][j]) {
|
||
|
|
t.Fatal(i, exp[i][j], got[i][j])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWrapNarrow(t *testing.T) {
|
||
|
|
exp := "The\nquick\nbrown\nfox\njumps\nover\nthe\nlazy\ndog."
|
||
|
|
if Wrap(text, 5) != exp {
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWrapOneLine(t *testing.T) {
|
||
|
|
exp := "The quick brown fox jumps over the lazy dog."
|
||
|
|
if Wrap(text, 500) != exp {
|
||
|
|
t.Fail()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWrapBug1(t *testing.T) {
|
||
|
|
cases := []struct {
|
||
|
|
limit int
|
||
|
|
text string
|
||
|
|
want string
|
||
|
|
}{
|
||
|
|
{4, "aaaaa", "aaaaa"},
|
||
|
|
{4, "a aaaaa", "a\naaaaa"},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, test := range cases {
|
||
|
|
got := Wrap(test.text, test.limit)
|
||
|
|
if got != test.want {
|
||
|
|
t.Errorf("Wrap(%q, %d) = %q want %q", test.text, test.limit, got, test.want)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|