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"}))