package onnx

import (
	"encoding/json"
	"os"
	"testing"
)

func TestH2ZBasic(t *testing.T) {
	tests := []struct {
		in   string
		want string
	}{
		{"abc123XYZ", "ａｂｃ１２３ＸＹＺ"},
		{"ｶﾞｯｺｳ", "ガッコウ"},
		{"ﾊﾟﾝ", "パン"},
		{"Hello, world!", "Ｈｅｌｌｏ，\u3000ｗｏｒｌｄ！"},
		{"「そんな……嘘でしょう！？」", "「そんな……嘘でしょう！？」"},
	}
	for _, tt := range tests {
		got := H2Z(tt.in)
		if got != tt.want {
			t.Errorf("H2Z(%q) = %q, want %q", tt.in, got, tt.want)
		}
	}
}

// testdata/jaconv_h2z.json is jaconv 0.5.0's h2z(s, ascii=True, digit=True),
// what manga-ocr applies, over ASCII, half-width kana with and without
// (han)dakuten, and the kana and full-width blocks it must leave alone.
func TestH2ZParityWithJaconv(t *testing.T) {
	data, err := os.ReadFile("testdata/jaconv_h2z.json")
	if err != nil {
		t.Fatal(err)
	}
	var m map[string]string
	if err := json.Unmarshal(data, &m); err != nil {
		t.Fatal(err)
	}
	for orig, want := range m {
		got := H2Z(orig)
		if got != want {
			t.Errorf("char %q (U+%04X): got %q, want %q", orig, []rune(orig)[0], got, want)
		}
	}
}

func TestIsKanaOnly(t *testing.T) {
	if !IsKanaOnly("あくま") {
		t.Errorf("expected あくま to be kana only")
	}
	if !IsKanaOnly("ラーメン！") {
		t.Errorf("expected ラーメン！ to be kana only")
	}
	if IsKanaOnly("悪魔") {
		t.Errorf("expected 悪魔 NOT to be kana only")
	}
	if IsKanaOnly("悪魔と契約する") {
		t.Errorf("expected 悪魔と契約する NOT to be kana only")
	}
}