mirror of
https://github.com/mii443/AzooKeyKanaKanjiConverter.git
synced 2025-08-22 15:05:26 +00:00
340 lines
16 KiB
Swift
340 lines
16 KiB
Swift
//
|
||
// DicdataStoreTests.swift
|
||
// azooKeyTests
|
||
//
|
||
// Created by ensan on 2023/02/09.
|
||
// Copyright © 2023 ensan. All rights reserved.
|
||
//
|
||
|
||
@testable import KanaKanjiConverterModule
|
||
@testable import KanaKanjiConverterModuleWithDefaultDictionary
|
||
import XCTest
|
||
|
||
final class DicdataStoreTests: XCTestCase {
|
||
func sequentialInput(_ composingText: inout ComposingText, sequence: String, inputStyle: KanaKanjiConverterModule.InputStyle) {
|
||
for char in sequence {
|
||
composingText.insertAtCursorPosition(String(char), inputStyle: inputStyle)
|
||
}
|
||
}
|
||
|
||
func requestOptions() -> ConvertRequestOptions {
|
||
.withDefaultDictionary(
|
||
N_best: 5,
|
||
requireJapanesePrediction: true,
|
||
requireEnglishPrediction: false,
|
||
keyboardLanguage: .ja_JP,
|
||
typographyLetterCandidate: false,
|
||
unicodeCandidate: true,
|
||
englishCandidateInRoman2KanaInput: true,
|
||
fullWidthRomanCandidate: false,
|
||
halfWidthKanaCandidate: false,
|
||
learningType: .nothing,
|
||
maxMemoryCount: 0,
|
||
shouldResetMemory: false,
|
||
memoryDirectoryURL: URL(fileURLWithPath: ""),
|
||
sharedContainerURL: URL(fileURLWithPath: ""),
|
||
metadata: nil
|
||
)
|
||
}
|
||
|
||
/// 絶対に変換できるべき候補をここに記述する
|
||
/// - 主に「変換できない」と報告のあった候補を追加する
|
||
func testMustWords() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
let mustWords = [
|
||
("アサッテ", "明後日"),
|
||
("オトトシ", "一昨年"),
|
||
("ダイヒョウ", "代表"),
|
||
("ヤマダ", "山田"),
|
||
("アイロ", "隘路"),
|
||
("フツカ", "二日"),
|
||
("フツカ", "2日"),
|
||
("ガデンインスイ", "我田引水"),
|
||
("フトウフクツ", "不撓不屈"),
|
||
("ナンタイ", "軟体"),
|
||
("ナンジ", "何時"),
|
||
("ナド", "等"),
|
||
// 各50音についてチェック（辞書の破損を調べるため）
|
||
("アイコウ", "愛好"),
|
||
("インガ", "因果"),
|
||
("ウンケイ", "運慶"),
|
||
("エンセキ", "縁石"),
|
||
("オンネン", "怨念"),
|
||
("カイビャク", "開闢"),
|
||
("ガンゼン", "眼前"),
|
||
("キトク", "奇特"),
|
||
("ギョウコ", "凝固"),
|
||
("クウキョ", "空虚"),
|
||
("グウワ", "寓話"),
|
||
("ケイセイ", "形声"),
|
||
("ゲントウ", "厳冬"),
|
||
("コウシャク", "講釈"),
|
||
("ゴリョウ", "御陵"),
|
||
("サンジュツ", "算術"),
|
||
("ザイアク", "罪悪"),
|
||
("ショウシャ", "瀟洒"),
|
||
("ジョウドウ", "情動"),
|
||
("スイサイ", "水彩"),
|
||
("ズイイ", "随意"),
|
||
("センカイ", "旋回"),
|
||
("ゼッカ", "舌禍"),
|
||
("ソツイ", "訴追"),
|
||
("ゾウゴ", "造語"),
|
||
("タイコウ", "太閤"),
|
||
("ダツリン", "脱輪"),
|
||
("チンコウ", "沈降"),
|
||
// ("ヂ")
|
||
("ツウショウ", "通商"),
|
||
// ("ヅ")
|
||
("テンキュウ", "天球"),
|
||
("デンシン", "伝心"),
|
||
("トウキ", "投機"),
|
||
("ドウモウ", "獰猛"),
|
||
("ナイシン", "内心"),
|
||
("ニンショウ", "人称"),
|
||
("ヌマヅ", "沼津"),
|
||
("ネンショウ", "燃焼"),
|
||
("ノウリツ", "能率"),
|
||
("ハクタイ", "百代"),
|
||
("バクシン", "驀進"),
|
||
("パク", "朴"),
|
||
("ヒショウ", "飛翔"),
|
||
("ビクウ", "鼻腔"),
|
||
("ピーシー", "PC"),
|
||
("フウガ", "風雅"),
|
||
("ブンジョウ", "分譲"),
|
||
("プラハノハル", "プラハの春"),
|
||
("ヘンリョウ", "変量"),
|
||
("ベイカ", "米価"),
|
||
("ペキン", "北京"),
|
||
("ホウトウ", "放蕩"),
|
||
("ボウダイ", "膨大"),
|
||
("ポリブクロ", "ポリ袋"),
|
||
("マッタン", "末端"),
|
||
("ミジン", "微塵"),
|
||
("ムソウ", "夢想"),
|
||
("メンツ", "面子"),
|
||
("モウコウ", "猛攻"),
|
||
("ヤクモノ", "約物"),
|
||
("ユウタイ", "有袋"),
|
||
("ヨウラン", "揺籃"),
|
||
("ランショウ", "濫觴"),
|
||
("リンネ", "輪廻"),
|
||
("ルイジョウ", "累乗"),
|
||
("レイラク", "零落"),
|
||
("ロウジョウ", "楼上"),
|
||
("ワクセイ", "惑星"),
|
||
("ヲ", "を"),
|
||
]
|
||
for (key, word) in mustWords {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition(key, inputStyle: .direct)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex), needTypoCorrection: false)
|
||
// 冗長な書き方だが、こうすることで「どの項目でエラーが発生したのか」がはっきりするため、こう書いている。
|
||
XCTAssertEqual(result.first(where: {$0.data.word == word})?.data.word, word)
|
||
}
|
||
}
|
||
|
||
/// 入っていてはおかしい候補をここに記述する
|
||
/// - 主に以前混入していたが取り除いた語を記述する
|
||
func testMustNotWords() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
let mustWords = [
|
||
("タイ", "体."),
|
||
("アサッテ", "明日"),
|
||
("チョ", "ちょwww"),
|
||
("シンコウホウホウ", "進行方向"),
|
||
("a", "あ"), // direct入力の場合「a」で「あ」をサジェストしてはいけない
|
||
("\\n", "\n")
|
||
]
|
||
for (key, word) in mustWords {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition(key, inputStyle: .direct)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex), needTypoCorrection: false)
|
||
XCTAssertNil(result.first(where: {$0.data.word == word && $0.data.ruby == key}))
|
||
}
|
||
}
|
||
|
||
/// 入力誤りを確実に修正できてほしい語群
|
||
func testMustCorrectTypo() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
let mustWords = [
|
||
("タイカクセイ", "大学生"),
|
||
("シヨック", "ショック"),
|
||
("キヨクイン", "局員"),
|
||
("シヨーク", "ジョーク"),
|
||
("サリカニ", "ザリガニ"),
|
||
("ノクチヒテヨ", "野口英世"),
|
||
("オタノフナカ", "織田信長"),
|
||
]
|
||
for (key, word) in mustWords {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition(key, inputStyle: .direct)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex), needTypoCorrection: true)
|
||
XCTAssertEqual(result.first(where: {$0.data.word == word})?.data.word, word)
|
||
}
|
||
}
|
||
|
||
/// 入力誤りを確実に修正できてほしい語群
|
||
func testMustCorrectTypoRoman2Kana() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
let mustWords = [
|
||
("tskamatsu", "高松"), // ts -> タ
|
||
("kitsmura", "北村"), // ts -> タ
|
||
]
|
||
for (key, word) in mustWords {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition(key, inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex), needTypoCorrection: true)
|
||
XCTAssertEqual(result.first(where: {$0.data.word == word})?.data.word, word)
|
||
}
|
||
}
|
||
|
||
func testLookupDicdata() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("ヘンカン", inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, 2 ..< 4))
|
||
XCTAssertFalse(result.contains(where: {$0.data.word == "変"}))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "変化"}))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "変換"}))
|
||
}
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("ヘンカン", inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, 0..<4))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "変"}))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "変化"}))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "変換"}))
|
||
}
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("ツカッ", inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, 2..<3))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "使っ"}))
|
||
}
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("ツカッt", inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, 2..<4))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "使っ"}))
|
||
}
|
||
do {
|
||
var c = ComposingText()
|
||
sequentialInput(&c, sequence: "tukatt", inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, 4..<6))
|
||
XCTAssertFalse(result.contains(where: {$0.data.word == "使っ"}))
|
||
}
|
||
do {
|
||
var c = ComposingText()
|
||
sequentialInput(&c, sequence: "tukatt", inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, surfaceRange: (0, nil))
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "使っ"}))
|
||
}
|
||
}
|
||
|
||
func testWiseDicdata() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("999999999999", inputStyle: .roman2kana)
|
||
let result = dicdataStore.getWiseDicdata(convertTarget: c.convertTarget, inputData: c, surfaceRange: 0..<12)
|
||
XCTAssertTrue(result.contains(where: {$0.word == "999999999999"}))
|
||
XCTAssertTrue(result.contains(where: {$0.word == "九千九百九十九億九千九百九十九万九千九百九十九"}))
|
||
}
|
||
}
|
||
|
||
func testDynamicUserDict() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
|
||
// 動的ユーザ辞書を設定
|
||
let testDynamicUserDict = [
|
||
DicdataElement(word: "テスト単語", ruby: "テストタンゴ", lcid: CIDData.固有名詞.cid, rcid: CIDData.固有名詞.cid, mid: MIDData.一般.mid, value: -10),
|
||
DicdataElement(word: "カスタム変換", ruby: "カスタムヘンカン", lcid: CIDData.固有名詞.cid, rcid: CIDData.固有名詞.cid, mid: MIDData.一般.mid, value: -12),
|
||
DicdataElement(word: "動的辞書", ruby: "ドウテキジショ", lcid: CIDData.固有名詞.cid, rcid: CIDData.固有名詞.cid, mid: MIDData.一般.mid, value: -11)
|
||
]
|
||
dicdataStore.sendToDicdataStore(.importDynamicUserDict(testDynamicUserDict))
|
||
|
||
// 完全一致テスト
|
||
do {
|
||
let result = dicdataStore.getMatchDynamicUserDict("テストタンゴ")
|
||
XCTAssertEqual(result.count, 1)
|
||
XCTAssertEqual(result.first?.word, "テスト単語")
|
||
XCTAssertEqual(result.first?.ruby, "テストタンゴ")
|
||
}
|
||
|
||
// 前方一致テスト
|
||
do {
|
||
let result = dicdataStore.getPrefixMatchDynamicUserDict("カスタム")
|
||
XCTAssertEqual(result.count, 1)
|
||
XCTAssertEqual(result.first?.word, "カスタム変換")
|
||
XCTAssertEqual(result.first?.ruby, "カスタムヘンカン")
|
||
}
|
||
|
||
// 変換動作テスト
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("テストタンゴ", inputStyle: .direct)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex), needTypoCorrection: false)
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "テスト単語"}))
|
||
}
|
||
|
||
// 複数の動的ユーザ辞書エントリの変換テスト
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("ドウテキジショ", inputStyle: .direct)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex), needTypoCorrection: false)
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "動的辞書"}))
|
||
}
|
||
|
||
// 存在しないエントリのテスト
|
||
do {
|
||
let result = dicdataStore.getMatchDynamicUserDict("ソンザイシナイ")
|
||
XCTAssertEqual(result.count, 0)
|
||
}
|
||
}
|
||
|
||
func testDynamicUserDictWithConversion() throws {
|
||
let dicdataStore = DicdataStore(convertRequestOptions: requestOptions())
|
||
|
||
// 動的ユーザ辞書を設定
|
||
let testDynamicUserDict = [
|
||
DicdataElement(word: "テストワード", ruby: "テストワード", lcid: CIDData.固有名詞.cid, rcid: CIDData.固有名詞.cid, mid: MIDData.一般.mid, value: -8),
|
||
DicdataElement(word: "特殊読み", ruby: "トクシュヨミ", lcid: CIDData.固有名詞.cid, rcid: CIDData.固有名詞.cid, mid: MIDData.一般.mid, value: -9)
|
||
]
|
||
dicdataStore.sendToDicdataStore(.importDynamicUserDict(testDynamicUserDict))
|
||
|
||
// ローマ字入力での変換テスト
|
||
do {
|
||
var c = ComposingText()
|
||
sequentialInput(&c, sequence: "tesutowaーdo", inputStyle: .roman2kana)
|
||
let result = dicdataStore.lookupDicdata(
|
||
composingText: c,
|
||
inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex),
|
||
surfaceRange: (0, c.convertTarget.count - 1 ..< c.convertTarget.count),
|
||
needTypoCorrection: false
|
||
)
|
||
XCTAssertTrue(result.contains(where: {$0.data.word == "テストワード"}))
|
||
XCTAssertEqual(result.first(where: {$0.data.word == "テストワード"})?.range, .surface(from: 0, to: 6))
|
||
}
|
||
|
||
// 動的ユーザ辞書の単語が通常の辞書よりも優先されることのテスト
|
||
do {
|
||
var c = ComposingText()
|
||
c.insertAtCursorPosition("トクシュヨミ", inputStyle: .direct)
|
||
let result = dicdataStore.lookupDicdata(composingText: c, inputRange: (0, c.input.endIndex - 1 ..< c.input.endIndex), needTypoCorrection: false)
|
||
let dynamicUserDictResult = result.first(where: {$0.data.word == "特殊読み"})
|
||
XCTAssertNotNil(dynamicUserDictResult)
|
||
XCTAssertEqual(dynamicUserDictResult?.data.metadata, .isFromUserDictionary)
|
||
}
|
||
}
|
||
|
||
func testPossibleNexts() throws {
|
||
let possibleNexts = InputStyleManager.shared.table(for: .defaultRomanToKana).possibleNexts
|
||
XCTAssertEqual(Set(possibleNexts["f", default: []]).symmetricDifference(["ファ", "フィ", "フ", "フェ", "フォ", "フャ", "フュ", "フョ", "フゥ", "ッf"]), [])
|
||
XCTAssertEqual(Set(possibleNexts["xy", default: []]).symmetricDifference(["ャ", "ョ", "ュ"]), [])
|
||
XCTAssertEqual(possibleNexts["", default: []], [])
|
||
}
|
||
}
|