secure/bidirule/go1_7_test.go

// Copyright 2016 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

// +build go1.7

package bidirule

import (
	"fmt"
	"testing"
)

func doTests(t *testing.T, fn func(t *testing.T, tc ruleTest)) {
	for rule, cases := range testCases {
		for i, tc := range cases {
			name := fmt.Sprintf("%d/%d:%+q:%s", rule, i, tc.in, tc.in)
			t.Run(name, func(t *testing.T) {
				fn(t, tc)
			})
		}
	}
}

var benchData = []struct{ name, data string }{
	{"ascii", "Scheveningen"},
	{"arabic", "دبي"},
	{"hangul", "다음과"},
}

func doBench(b *testing.B, fn func(b *testing.B, data string)) {
	for _, d := range benchData {
		b.Run(d.name, func(b *testing.B) { fn(b, d.data) })
	}
}

func BenchmarkSpan(b *testing.B) {
	r := New()
	doBench(b, func(b *testing.B, str string) {
		b.SetBytes(int64(len(str)))
		data := []byte(str)
		for i := 0; i < b.N; i++ {
			r.Reset()
			r.Span(data, true)
		}
	})
}

func BenchmarkDirectionASCII(b *testing.B) {
	doBench(b, func(b *testing.B, str string) {
		b.SetBytes(int64(len(str)))
		data := []byte(str)
		for i := 0; i < b.N; i++ {
			Direction(data)
		}
	})
}

func BenchmarkDirectionStringASCII(b *testing.B) {
	doBench(b, func(b *testing.B, str string) {
		b.SetBytes(int64(len(str)))
		for i := 0; i < b.N; i++ {
			DirectionString(str)
		}
	})
}