def test_persist(self):
copy_tst_data("I_1_two_entries", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
register._lemmas[0]._lemma_dict["previous"] = None
register._lemmas[0]._chapters[0]._dict["author"] = "ÄäÖöÜüß"
register.persist()
expect = """[
{
"lemma": "Aal",
"next": "Aarassos",
"proof_read": 3,
"short_description": "Ein Fisch",
"chapters": [
{
"start": 1,
"end": 4,
"author": "ÄäÖöÜüß"
}
]
},
{
"lemma": "Aarassos",
"previous": "Aal",
"proof_read": 2,
"chapters": [
{
"start": 4,
"end": 4,
"author": "Abert"
}
]
}
]"""
with open(DataRepo.get_data_path().joinpath("I_1.json"), mode="r", encoding="utf-8") as register_file:
compare(expect, register_file.read())
def setUp(self):
copy_tst_data("I_1_alpha", "I_1")
copy_tst_data("III_1_alpha", "III_1")
self.authors = Authors()
self.volumes = Volumes()
self.registers = OrderedDict()
self.registers["I,1"] = VolumeRegister(self.volumes["I,1"], self.authors)
self.registers["III,1"] = VolumeRegister(self.volumes["III,1"], self.authors)
def test_header_vg_nf(self):
copy_tst_data("I_1_base", "I_1")
i1 = VolumeRegister(Volumes()["I,1"], Authors())._get_header()
self.assertTrue("VG=" in i1)
self.assertTrue("NF=I,2" in i1)
copy_tst_data("I_1_base", "S I")
si = VolumeRegister(Volumes()["S I"], Authors())._get_header()
self.assertTrue("VG=X A" in si)
self.assertTrue("NF=S II" in si)
def test_update_lemma(self):
copy_tst_data("I_1_base", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "Aal", "redirect": True}
with Updater(register) as updater:
updater.update_lemma(update_dict, ["next"])
post_lemma = register.get_lemma_by_name("Aal")
self.assertTrue(post_lemma["redirect"])
self.assertIsNone(post_lemma["next"])
def test_update_next_and_previous(self):
copy_tst_data("I_1_sorting2", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "O", "previous": "Ä", "next": "Ü"}
with Updater(register) as updater:
updater._try_update_next_and_previous(update_dict, register.get_lemma_by_name("O"))
post_lemma_previous = register.get_lemma_by_name("Ä")
compare("O", post_lemma_previous["next"])
post_lemma_next = register.get_lemma_by_name("Ü")
compare("O", post_lemma_next["previous"])
def test_update_by_sortkey_raise_error(self):
copy_tst_data("I_1_update_previous_wrong", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "Äarassos", "previous": "Aal", "next": "Aba 1", "sort_key": "Aarassos"}
with self.assertRaisesRegex(RegisterException, "!= next lemma name \"Ab 1\""):
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
previous_lemma = register.get_lemma_by_name("Aal")
compare("Aarassos", previous_lemma["next"])
next_lemma = register.get_lemma_by_name("Ab 1")
compare("Aarassos", next_lemma["previous"])
def test_update_lemma_by_sortkey(self):
copy_tst_data("I_1_base", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "Äal", "redirect": True, "sort_key": "Aal", "next": "Aarassos"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
post_lemma = register.get_lemma_by_name("Äal")
compare(True, post_lemma["redirect"])
compare("Aal", post_lemma["sort_key"])
post_lemma_next = register.get_lemma_by_name("Aarassos")
compare("Äal", post_lemma_next["previous"])
def test_duplicate_lemmas_in_supplements(self):
copy_tst_data("S_I_no_dublicates", "S I")
register = VolumeRegister(Volumes()["S I"], Authors())
update_dict = {"lemma": "Abdymon", "previous": "Abd Hadad", "next": "Abeikta"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
compare(11, len(register.lemmas))
def test_update_pre_and_next_not_possible(self):
copy_tst_data("I_1_sorting2", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "B", "previous": "A", "next": "D"}
with self.assertRaisesRegex(RegisterException, "Diff between previous and next aren't 1 or 2"):
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
def test_get_lemma_by_id(self):
copy_tst_data("I_1_base", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
lemma = register[2]
compare("Aarassos", lemma["previous"])
with self.assertRaises(IndexError):
print(register[8])
def test_header_proof_read(self):
copy_tst_data("I_1_base", "I_1")
i1 = VolumeRegister(Volumes()["I,1"], Authors())._get_header().replace("\n", "")
compare(StringComparison(".*SUM=8.*"), i1)
compare(StringComparison(".*UNK=3.*"), i1)
compare(StringComparison(".*KOR=2.*"), i1)
compare(StringComparison(".*FER=3.*"), i1)
def update_lemma(self,
lemma_dict: LemmaDict,
remove_items: UpdaterRemoveList,
self_supplement: bool = False) -> str:
sort_key = VolumeRegister.normalize_sort_key(lemma_dict)
if "lemma" in lemma_dict and self._register.get_lemma_by_name(
lemma_dict["lemma"], self_supplement):
self._update_lemma_by_name(lemma_dict, remove_items,
self_supplement)
return "update_lemma_by_name"
if self._register.get_lemma_by_sort_key(sort_key):
self._update_by_sortkey(lemma_dict, remove_items)
return "update_by_sortkey"
if "previous" in lemma_dict and "next" in lemma_dict \
and self._register.get_lemma_by_sort_key(Lemma.make_sort_key(lemma_dict["previous"])) \
and self._register.get_lemma_by_sort_key(Lemma.make_sort_key(lemma_dict["next"])):
self._update_pre_and_post_exists(lemma_dict)
return "update_pre_and_post_exists"
if "previous" in lemma_dict \
and self._register.get_lemma_by_sort_key(Lemma.make_sort_key(lemma_dict["previous"])):
self._update_pre_exists(lemma_dict)
return "update_pre_exists"
if "next" in lemma_dict \
and self._register.get_lemma_by_sort_key(Lemma.make_sort_key(lemma_dict["next"])):
self._update_post_exists(lemma_dict)
return "update_post_exists"
raise RegisterException(
f"The update of the register {self._register.volume.name} "
f"with the dict {lemma_dict} is not possible. "
f"No strategy available")
def test_get_id_of_lemma(self):
copy_tst_data("I_1_self_append", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
compare(0, register.get_index_of_lemma("Aal"))
compare(2, register.get_index_of_lemma("Something"))
compare(3, register.get_index_of_lemma("Aal", self_supplement=True))
lemma = register.get_lemma_by_name("Aal", self_supplement=True)
compare(3, register.get_index_of_lemma(lemma))
compare(None, register.get_index_of_lemma("Lemma not there"))
def test_prokleides_2(self):
copy_tst_data("prokleides_2_bug", "XXIII_1")
register = VolumeRegister(Volumes()["XXIII,1"], Authors())
update_dict = {"lemma": "Prokleides 2", "previous": "Prokleides", "next": "Prokles"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
compare(None, register.lemmas[0]["previous"])
compare(None, register.lemmas[2]["next"])
def test_bug_to_much_percent(self):
copy_tst_data("I_1_author_bug_percent", "I_1")
copy_tst_data("III_1_author_bug_percent", "III_1")
registers = OrderedDict()
registers["I,1"] = VolumeRegister(self.volumes["I,1"], self.authors)
registers["III,1"] = VolumeRegister(self.volumes["III,1"],
self.authors)
abel_register = AuthorRegister(self.authors.get_author("Herman Abel"),
self.authors, registers)
compare(
"|-\n"
"|data-sort-value=\"Abel, Herman\""
"|[[Paulys Realencyclopädie der classischen Altertumswissenschaft/Register/Herman Abel|Herman Abel]]\n"
"|data-sort-value=\"0005\"|5\n"
"|data-sort-value=\"100.0\"|100.0%\n"
"|<span style=\"color:#669966\">████████████████████</span>"
"<span style=\"color:#556B2F\"></span>"
"<span style=\"color:#AA0000\"></span>",
abel_register.overview_line)
def test_update_no_update_possible(self):
copy_tst_data("I_1_base", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "bubum",
"redirect": True,
"sort_key": "babam",
"previous": "rubbish",
"next": "something"}
with self.assertRaisesRegex(RegisterException, "No strategy available"):
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
def test_acutius_1a(self):
copy_tst_data("acutius_1a_bug", "S I")
register = VolumeRegister(Volumes()["S I"], Authors())
update_dict = {"lemma": "Acutius a", "previous": "Acronoma", "next": "Acutius 1a", "sort_key": "Acutius 0a"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
update_dict = {"lemma": "Acutius 1a", "previous": "Acutius a", "next": "Adaba", "redirect": True}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
# don't create a new lemma, because Acutius a != Acutius 0a
compare(5, len(register.lemmas))
def __init__(self, update_data=False):
self.repo = DataRepo()
if update_data:
self.repo.pull()
self._authors: Authors = Authors()
self._registers: Dict[str, VolumeRegister] = OrderedDict()
self._alphabetic_registers: Dict[str,
AlphabeticRegister] = OrderedDict()
for volume in Volumes().all_volumes:
with contextlib.suppress(FileNotFoundError):
self._registers[volume.name] = VolumeRegister(
volume, self._authors)
def test_update_create_next_previous_supplement_by_name_next_exists(self):
copy_tst_data("I_1_sorting2", "R")
register = VolumeRegister(Volumes()["R"], Authors())
update_dict = {"lemma": "O", "previous": "N", "next": "Ü"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
post_lemma = register.get_lemma_by_name("O")
compare("N", post_lemma["previous"])
compare("Ü", post_lemma["next"])
post_lemma_previous = register.get_lemma_by_name("N")
compare("O", post_lemma_previous["next"])
post_lemma_previous_previous = register.get_lemma_by_name("A")
compare(None, post_lemma_previous_previous["next"])
post_lemma_next = register.get_lemma_by_name("Ü")
compare("O", post_lemma_next["previous"])
post_lemma_next_next = register.get_lemma_by_name("D")
compare("Ü", post_lemma_next_next["previous"])
self.assertTrue(register.get_index_of_lemma("A") <
register.get_index_of_lemma("N") <
register.get_index_of_lemma("O") <
register.get_index_of_lemma("Ü") <
register.get_index_of_lemma("D"))
def test_get_table(self):
copy_tst_data("I_1_two_entries", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
expected_table = """{{RERegister
|BAND=I,1
|VG=
|NF=I,2
|SUM=2
|UNK=1
|KOR=1
|FER=0
}}
{|class="wikitable sortable"
!Artikel
!Kurztext
!Wikilinks
!Seite
!Autor
!Stat
|-
|data-sort-value="aal"|[[RE:Aal|'''{{Anker2|Aal}}''']]
||Ein Fisch
||
|[https://elexikon.ch/meyers/RE/I,1_1.png 1]-4
|Herman Abel
|style="background:#FFCBCB"|2069
|-
|data-sort-value="aarassos"|[[RE:Aarassos|'''{{Anker2|Aarassos}}''']]
||
||
|[https://elexikon.ch/meyers/RE/I,1_5.png 4]
|Abert
|style="background:#556B2F"|KOR
|}
[[Kategorie:RE:Register|!]]"""
compare(expected_table, register.get_register_str())
def test_get_lemma_self_append(self):
copy_tst_data("I_1_self_append", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
lemma = register.get_lemma_by_name("Aal")
compare(None, lemma["previous"])
lemma = register.get_lemma_by_name("Aal", self_supplement=True)
compare("Something", lemma["previous"])
lemma = register.get_lemma_by_sort_key("AAL", self_supplement=True)
compare("Something", lemma["previous"])
lemma = register.get_lemma_by_name("Something", self_supplement=True)
compare(None, lemma)
def test_update_lemma_by_sortkey_pre_and_next_lemma_other_name(self):
copy_tst_data("I_1_sorting2", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "Ö", "sort_key": "O", "previous": "Ä", "next": "Ü"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
post_lemma = register.get_lemma_by_name("Ö")
compare("O", post_lemma["sort_key"])
post_lemma_previous = register.get_lemma_by_name("Ä")
compare("Ö", post_lemma_previous["next"])
post_lemma_next = register.get_lemma_by_name("Ü")
compare("Ö", post_lemma_next["previous"])
post_lemma_start = register.get_lemma_by_name("Vor A")
compare("Ä", post_lemma_start["next"])
post_lemma_end = register.get_lemma_by_name("D")
compare("Ü", post_lemma_end["previous"])
def test_update_by_replace(self):
copy_tst_data("I_1_sorting2", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "B", "previous": "Ä", "next": "Ü"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
post_lemma = register.get_lemma_by_name("B")
compare("Ä", post_lemma["previous"])
compare("Ü", post_lemma["next"])
post_lemma_previous = register.get_lemma_by_name("Ä")
compare("B", post_lemma_previous["next"])
post_lemma_previous_previous = register.get_lemma_by_name("Vor A")
compare("Ä", post_lemma_previous_previous["next"])
post_lemma_next = register.get_lemma_by_name("Ü")
compare("B", post_lemma_next["previous"])
post_lemma_next_next = register.get_lemma_by_name("D")
compare("Ü", post_lemma_next_next["previous"])
def test_update_by_insert_before_next_no_previous(self):
copy_tst_data("I_1_sorting2", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
update_dict = {"lemma": "B", "next": "Ö"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
post_lemma = register.get_lemma_by_name("B")
compare(None, post_lemma["previous"])
compare("Ö", post_lemma["next"])
post_lemma_previous = register.get_lemma_by_name("A")
compare(None, post_lemma_previous["next"])
post_lemma_next = register.get_lemma_by_name("Ö")
compare("B", post_lemma_next["previous"])
post_lemma_next_next = register.get_lemma_by_name("U")
compare("Ö", post_lemma_next_next["previous"])
def test_update_create_next_previous_supplement_by_sort_key(self):
copy_tst_data("I_1_sorting2", "S I")
register = VolumeRegister(Volumes()["S I"], Authors())
update_dict = {"lemma": "Ö", "previous": "blub", "next": "Ä"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
compare(8, len(register.lemmas))
update_dict = {"lemma": "Ä", "previous": "Ö", "next": "blab"}
with Updater(register) as updater:
updater.update_lemma(update_dict, [])
compare(9, len(register.lemmas))
post_lemma = register.get_lemma_by_name("Ö")
compare("blub", post_lemma["previous"])
compare("Ä", post_lemma["next"])
post_lemma = register.get_lemma_by_name("Ä")
compare("Ö", post_lemma["previous"])
compare("blab", post_lemma["next"])
self.assertTrue(register.get_index_of_lemma("Ä") <
register.get_index_of_lemma("blab") <
register.get_index_of_lemma("blub") <
register.get_index_of_lemma("Ö"))
def test_header_band(self):
copy_tst_data("I_1_base", "I_1")
self.assertTrue("BAND=I,1" in VolumeRegister(Volumes()["I,1"], Authors())._get_header())
copy_tst_data("I_1_base", "S I")
self.assertTrue("BAND=S I" in VolumeRegister(Volumes()["S I"], Authors())._get_header())
def test_json_integrity(self): # pragma: no cover
for volume in Volumes().all_volumes:
print(volume)
VolumeRegister(volume, Authors())
def test_init(self):
copy_tst_data("I_1_base", "I_1")
VolumeRegister(Volumes()["I,1"], Authors())
def test_get_lemma_by_sort_key(self):
copy_tst_data("I_1_base", "I_1")
register = VolumeRegister(Volumes()["I,1"], Authors())
lemma = register.get_lemma_by_sort_key("äbÄ 1")
compare("Aarassos", lemma["previous"])
self.assertIsNone(register.get_lemma_by_name("Abracadabra"))
def test_normalize_sort_key(self):
compare("aaa", VolumeRegister.normalize_sort_key({"lemma": "ååå"}))
compare("bbb", VolumeRegister.normalize_sort_key({"lemma": "ååå", "sort_key": "bbb"}))
