generator_test.go

package generator

import "testing"

func TestStart1(t *testing.T) {
	config := Config{}
	actual, err := Start(config)
	t.Log(actual)
	if err != nil {
		t.Errorf("handler returned unexpected body: got %v", err)
	}
}

func TestStart2(t *testing.T) {
	config := Config{
		PunctuationLevel: 4,
	}
	_, err := Start(config)
	t.Log(err)
	if err == nil {
		t.Errorf("handler returned unexpected body: got nil")
	}
}

func TestKatakanaKatsuyou1(t *testing.T) {
	expected := "なんちゃッテ"
	actual := katakanaKatsuyou("なんちゃって", 2)
	t.Log(actual)
	if actual != expected {
		t.Errorf("handler returned unexpected body: got %v want %v", actual, expected)
	}
}

func TestKatakanaKatsuyou2(t *testing.T) {
	expected := "どうしちゃったノカナ{EMOJI_POS}"
	actual := katakanaKatsuyou("どうしちゃったのかな{EMOJI_POS}", 3)
	t.Log(actual)
	if actual != expected {
		t.Errorf("handler returned unexpected body: got %v want %v", actual, expected)
	}
}

func TestKatakanaKatsuyou3(t *testing.T) {
	expected := "どうしちゃったのかな{EMOJI_POS}"
	actual := katakanaKatsuyou("どうしちゃったのかな{EMOJI_POS}", 0)
	t.Log(actual)
	if actual != expected {
		t.Errorf("handler returned unexpected body: got %v want %v", actual, expected)
	}
}

func TestKatakanaKatsuyou4(t *testing.T) {
	expected := "東西南北{EMOJI_POS}"
	actual := katakanaKatsuyou("東西南北{EMOJI_POS}", 2)
	t.Log(actual)
	if actual != expected {
		t.Errorf("handler returned unexpected body: got %v want %v", actual, expected)
	}
}

func TestInsertPunctuations0(t *testing.T) {
	expected := "どうしちゃったのかな"
	actual := insertPunctuations("どうしちゃったのかな", PunctuationConfig{[]string{"助動詞", "名詞"}, 0})
	t.Log(actual)
	if actual != expected {
		t.Errorf("handler returned unexpected body: got %v want %v", actual, expected)
	}
}

func TestInsertPunctuations2(t *testing.T) {
	expected := "どうしちゃった、のかな"
	actual := insertPunctuations("どうしちゃったのかな", PunctuationConfig{[]string{"助動詞"}, 100})
	t.Log(actual)
	if actual != expected {
		t.Errorf("handler returned unexpected body: got %v want %v", actual, expected)
	}
}
func TestInsertPunctuations3(t *testing.T) {
	expected := "どうしちゃった、のか、な、"
	actual := insertPunctuations("どうしちゃったのかな", PunctuationConfig{[]string{"助動詞", "助詞"}, 100})
	t.Log(actual)
	if actual != expected {
		t.Errorf("handler returned unexpected body: got %v want %v", actual, expected)
	}
}

func BenchmarkPunctuation(b *testing.B) {
	for i:=0; i < b.N; i++ {
		insertPunctuations("どうしちゃったのかな", PunctuationConfig{[]string{"助動詞", "助詞"}, 100})
	}
}