package textproc

import "testing"

func TestPostProcessRemovesSpacesBetweenCJK(t *testing.T) {
	got := PostProcess("请 你 根 据 官 网")
	want := "请你根据官网"
	if got != want {
		t.Fatalf("PostProcess() = %q, want %q", got, want)
	}
}

func TestPostProcessKeepsSpaceBetweenCJKAndEnglish(t *testing.T) {
	got := PostProcess("请 你 open AI")
	want := "请你 open AI"
	if got != want {
		t.Fatalf("PostProcess() = %q, want %q", got, want)
	}
}

func TestPostProcessKeepsEnglishWordSpaces(t *testing.T) {
	got := PostProcess("open source speech recognition")
	want := "Open source speech recognition"
	if got != want {
		t.Fatalf("PostProcess() = %q, want %q", got, want)
	}
}