pinyingo_test.go

package pinyingo

import (
	"testing"
)

func TestConvert(t *testing.T) {
	pinyinMap := map[string][]string{
		"hello中国": []string{"hello", "zhōng", "guó"},
		"中国hello": []string{"zhōng", "guó", "hello"},
		"中国":      []string{"zhōng", "guó"},
		"123qwe":  []string{"123qwe"},
	}
	pinyinMap1 := map[string][]string{
		"hello中国": []string{"hello", "zh", "g"},
		"中国":      []string{"zh", "g"},
		"123qwe":  []string{"123qwe"},
	}
	pinyinMap2 := map[string][]string{
		"hello中国": []string{"hello", "zhong", "guo"},
		"中国":      []string{"zhong", "guo"},
		"123qwe":  []string{"123qwe"},
	}

	pinyinMap3 := map[string][]string{
		"重阳": []string{"chóng", "yáng"},
		"重点": []string{"zhòng", "diǎn"},
	}

	pinyinMap4 := map[string][]string{
		"你好啊": []string{"n", "h", "a"},
	}

	for k, v := range pinyinMap {
		py := NewPy(STYLE_TONE, NO_SEGMENT)
		converted := py.Convert(k)
		for i := 0; i < len(converted); i++ {
			if converted[i] != v[i] {
				t.Errorf("%s is not equal %s", converted, v)
			}
		}
	}

	for k, v := range pinyinMap1 {
		py := NewPy(STYLE_INITIALS, NO_SEGMENT)
		converted := py.Convert(k)
		for i := 0; i < len(converted); i++ {
			if converted[i] != v[i] {
				t.Errorf("%s is not equal %s", converted, v)
			}
		}
	}

	for k, v := range pinyinMap2 {
		py := NewPy(STYLE_NORMAL, NO_SEGMENT)
		converted := py.Convert(k)
		for i := 0; i < len(converted); i++ {
			if converted[i] != v[i] {
				t.Errorf("%s is not equal %s", converted, v)
			}
		}
	}

	for k, v := range pinyinMap3 {
		py := NewPy(STYLE_TONE, USE_SEGMENT)
		converted := py.Convert(k)
		for i := 0; i < len(converted); i++ {
			if converted[i] != v[i] {
				t.Errorf("%s is not equal %s", converted, v)
			}
		}
	}

	for k, v := range pinyinMap4 {
		py := NewPy(STYLE_FIRST_LETTER, USE_SEGMENT)
		converted := py.Convert(k)
		for i := 0; i < len(converted); i++ {
			if converted[i] != v[i] {
				t.Errorf("%s is not equal %s", converted, v)
			}
		}
	}
}