ngender_test.go

package ngender

import (
	"math"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestGuess(t *testing.T) {
	tolerance := .000001
	gender, probability := Guess("赵本山")
	assert.Equal(t, "male", gender)
	assert.True(t, math.Abs(probability-0.983622) < tolerance)

	gender, probability = Guess("宋丹丹")
	assert.Equal(t, "female", gender)
	assert.True(t, math.Abs(probability-0.975948) < tolerance)

	gender, probability = Guess("诸葛亮")
	assert.Equal(t, "male", gender)

	gender, probability = Guess("欧阳锋")
	assert.Equal(t, "male", gender)

	gender, probability = Guess("James")
	assert.Equal(t, "unknown", gender)

	gender, probability = Guess("Tinky Winky")
	assert.Equal(t, "unknown", gender)
}

func TestGuessGivenName(t *testing.T) {
	tolerance := .000001
	gender, probability := GuessGivenName("丹丹")
	assert.Equal(t, "female", gender)
	assert.True(t, math.Abs(probability-0.975948) < tolerance)

	gender, probability = GuessGivenName("俊杰")
	assert.Equal(t, "male", gender)

	gender, probability = GuessGivenName("晓娟")
	assert.Equal(t, "female", gender)

	gender, probability = GuessGivenName("James")
	assert.Equal(t, "unknown", gender)

	gender1, probability1 := Guess("司马懿")
	gender2, probability2 := GuessGivenName("懿")
	assert.Equal(t, gender1, gender2)
	assert.Equal(t, probability1, probability2)
}

func BenchmarkGuess(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, _ = Guess("王伟")
	}
}

func BenchmarkGuessGivenName(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_, _ = Guess("建国")
	}
}