naturally_test.go

package naturally_test

import (
	"github.com/dangogh/naturally"
	"sort"
	"testing"
)

var testdata0 = [...]string{"A1", "A0", "A21", "A11", "A111", "A21", "A2"}
var testexp0 = [...]string{"A0", "A1", "A2", "A11", "A21", "A21", "A111"}

// numeric only
var testdata1 = [...]string{"001", "2", "xyzzy", "30", "22", "0", "00", "3", "1"}
var testexp1 = [...]string{"0", "00", "1", "001", "2", "3", "22", "30", "xyzzy"}

var testdata2 = [...]string{"A1BA1", "A11AA1", "A2AB0", "B1AA1", "A1AA1"}
var testexp2 = [...]string{"A1AA1", "A1BA1", "A2AB0", "A11AA1", "B1AA1"}

// non-numeric only
var testdata3 = [...]string{"ZZ", "XX", "AAA", "A", "AA", "Z", "XX"}
var testexp3 = [...]string{"A", "AA", "AAA", "XX", "XX", "Z", "ZZ"}

func _testSorts(t *testing.T, data, expected []string) {
	n := naturally.StringSlice(data[0:])
	sort.Sort(n)

	for ii, got := range data {
		exp := expected[ii]
		if got != exp {
			t.Errorf("Got %v; expected %v\n", got, exp)

		}
	}
}

func TestSortsA(t *testing.T) {
	_testSorts(t, testdata0[0:], testexp0[0:])
}

func TestSortsB(t *testing.T) {
	_testSorts(t, testdata1[0:], testexp1[0:])
}

func TestSortsC(t *testing.T) {
	_testSorts(t, testdata2[0:], testexp2[0:])
}

func TestSortsD(t *testing.T) {
	_testSorts(t, testdata3[0:], testexp3[0:])
}

func BenchmarkSortNaturally(b *testing.B) {
	b.StopTimer()
	for ii := 0; ii < b.N; ii++ {
		n := naturally.StringSlice(testdata2[0:])
		b.StartTimer()
		sort.Sort(n)
		b.StopTimer()
	}
}

func BenchmarkSort(b *testing.B) {
	b.StopTimer()
	for ii := 0; ii < b.N; ii++ {
		n := sort.StringSlice(testdata2[0:])
		b.StartTimer()
		sort.Sort(n)
		b.StopTimer()
	}
}