144 lines
4.7 KiB
Python
144 lines
4.7 KiB
Python
#!/usr/bin/env python3
|
||
"""Unit tests for WolfDawn names.json helpers in util/wolf_names.py."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import os
|
||
import sys
|
||
import tempfile
|
||
import unittest
|
||
from pathlib import Path
|
||
|
||
ROOT = Path(__file__).resolve().parents[1]
|
||
os.chdir(ROOT)
|
||
sys.path.insert(0, str(ROOT))
|
||
|
||
from util import wolf_names as wn # noqa: E402
|
||
|
||
|
||
class TestNameTranslatable(unittest.TestCase):
|
||
def test_safe_and_refs_are_translatable(self):
|
||
self.assertTrue(wn.is_name_translatable({"safety": "safe"}))
|
||
self.assertTrue(wn.is_name_translatable({"safety": "refs"}))
|
||
|
||
def test_verify_and_missing_are_not_translatable(self):
|
||
self.assertFalse(wn.is_name_translatable({"safety": "verify"}))
|
||
self.assertFalse(wn.is_name_translatable({}))
|
||
self.assertFalse(wn.is_name_translatable({"safety": ""}))
|
||
|
||
|
||
class TestVocabHarvestCandidate(unittest.TestCase):
|
||
def test_short_weapon_name_harvests(self):
|
||
entry = {"source": "ダガー", "note": "武器", "safety": "safe"}
|
||
self.assertTrue(wn.is_vocab_harvest_candidate(entry))
|
||
|
||
def test_multiline_profile_skipped(self):
|
||
entry = {
|
||
"source": "セルリアと申します。\nよろしくお願いいたします。",
|
||
"note": "├■プロフィール",
|
||
"safety": "safe",
|
||
}
|
||
self.assertFalse(wn.is_vocab_harvest_candidate(entry))
|
||
|
||
def test_profile_note_skipped_even_when_single_line(self):
|
||
entry = {"source": "ローザだ。", "note": "├■プロフィール", "safety": "safe"}
|
||
self.assertFalse(wn.is_vocab_harvest_candidate(entry))
|
||
|
||
def test_resistance_label_still_harvests(self):
|
||
entry = {"source": "物理50％軽減", "note": "┣ 属性耐性", "safety": "safe"}
|
||
self.assertTrue(wn.is_vocab_harvest_candidate(entry))
|
||
|
||
|
||
class TestCountNameSafety(unittest.TestCase):
|
||
DOC = {
|
||
"kind": "names",
|
||
"names": [
|
||
{"source": "a", "safety": "safe"},
|
||
{"source": "b", "safety": "refs"},
|
||
{"source": "c", "safety": "verify"},
|
||
{"source": "d"},
|
||
],
|
||
}
|
||
|
||
def test_counts_badges(self):
|
||
counts = wn.count_name_safety(self.DOC)
|
||
self.assertEqual(counts["safe"], 1)
|
||
self.assertEqual(counts["refs"], 1)
|
||
self.assertEqual(counts["verify"], 1)
|
||
self.assertEqual(counts["unknown"], 1)
|
||
self.assertEqual(counts["translatable"], 2)
|
||
|
||
def test_summary_mentions_translatable_count(self):
|
||
summary = wn.format_name_safety_summary(self.DOC)
|
||
self.assertIn("2 of 4", summary)
|
||
self.assertIn("verify", summary)
|
||
|
||
|
||
class TestNoteHeader(unittest.TestCase):
|
||
def test_static_fallback_is_bilingual(self):
|
||
self.assertEqual(wn.note_header("武器"), "Weapon · 武器")
|
||
|
||
def test_live_db_label_wins_over_static(self):
|
||
labels = {"武器": "Blade"}
|
||
self.assertEqual(wn.note_header("武器", labels), "Blade · 武器")
|
||
|
||
def test_unknown_note_stays_japanese_only(self):
|
||
self.assertEqual(wn.note_header("■MOBセリフ"), "■MOBセリフ")
|
||
|
||
|
||
class TestCollectNameNotes(unittest.TestCase):
|
||
DOC = {
|
||
"kind": "names",
|
||
"names": [
|
||
{"source": "a", "note": "武器"},
|
||
{"source": "b", "note": "技能"},
|
||
{"source": "c", "note": "武器"},
|
||
{"source": "d"},
|
||
],
|
||
}
|
||
|
||
def test_collects_unique_sorted_notes(self):
|
||
self.assertEqual(wn.collect_name_notes(self.DOC), ["技能", "武器"])
|
||
|
||
def test_parse_json_notes(self):
|
||
self.assertEqual(
|
||
wn.parse_name_wrap_notes('["武器","├■プロフィール"]'),
|
||
frozenset({"武器", "├■プロフィール"}),
|
||
)
|
||
|
||
def test_parse_comma_separated_notes(self):
|
||
self.assertEqual(
|
||
wn.parse_name_wrap_notes("武器,技能"),
|
||
frozenset({"武器", "技能"}),
|
||
)
|
||
|
||
|
||
class TestDeriveDbLabels(unittest.TestCase):
|
||
def setUp(self):
|
||
self._tmp = tempfile.TemporaryDirectory()
|
||
self.dir = Path(self._tmp.name)
|
||
|
||
def tearDown(self):
|
||
self._tmp.cleanup()
|
||
|
||
def test_parses_bilingual_typenames_from_db_files(self):
|
||
(self.dir / "DataBase.project.json").write_text(
|
||
json.dumps({
|
||
"kind": "db",
|
||
"groups": [
|
||
{"typeName": "Weapon · 武器", "lines": []},
|
||
{"typeName": "Skill · 技能", "lines": []},
|
||
],
|
||
}),
|
||
encoding="utf-8",
|
||
)
|
||
labels = wn.derive_db_labels(self.dir)
|
||
self.assertEqual(labels, {"武器": "Weapon", "技能": "Skill"})
|
||
|
||
def test_missing_dir_returns_empty(self):
|
||
self.assertEqual(wn.derive_db_labels(self.dir / "nope"), {})
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|