Homec4science
Diffusion nietzsche-python Browse

nietzsche-python/tests_svgscripts/test_interactive_merger.py
rNIETZSCHEPYTHON87

Last Change
Raw File

test_interactive_merger.py
View Options

import unittest
from os import sep, path, remove
from os.path import isdir, isfile, dirname
import shutil
import sys
import lxml.etree as ET
import warnings
import sys
sys.path.append('svgscripts')
import compare_faksimile_words_line_wise
import interactive_merger
from datatypes.faksimile import FaksimilePage
from datatypes.page import Page
from datatypes.faksimile_position import FaksimilePosition
from datatypes.word_position import WordPosition
from datatypes.word import Word
class TestInteractiveMerger(unittest.TestCase):
def setUp(self):
compare_faksimile_words_line_wise.UNITTESTING = True
interactive_merger.UNITTESTING = True
DATADIR = path.dirname(__file__) + sep + 'test_data'
self.faksimile_dir = DATADIR + sep + 'faksimile_svg'
self.manuscript = DATADIR + sep + 'N_VII_1.xml'
self.manuscript_copy = self.manuscript.replace('.', '_copy.')
self.faksimile_file = self.faksimile_dir + sep + 'N-VII-1,5et6.svg'
self.xml_file = DATADIR + sep + 'N_VII_1_page005.xml'
self.Mp_XIV_1_mytest_421 = DATADIR + sep + 'Mp_XIV_1_mytest_421.xml'
self.correction_dir = DATADIR + sep + 'correction_dir'
self.page138 = DATADIR + sep + 'N_VII_1_page138.xml'
def test_mark_unmergeable_words_and_faksimile_positions(self):
page = Page(self.xml_file)
faksimile_page = FaksimilePage.get_faksimile_pages(self.faksimile_file, page_number=page.number)[0]
compare_faksimile_words_line_wise.mark_unmergeable_words_and_faksimile_positions(page.words, faksimile_page.word_positions)
word_gar = [ fp for fp in faksimile_page.word_positions if fp.text == 'gar' ]
self.assertTrue(len(word_gar) == 1)
self.assertFalse(word_gar[0].mergeable)
#print([word.text for word in page.words if not word.mergeable])
#print([fp.text for fp in faksimile_page.word_positions if not fp.mergeable])
def test_get_line(self):
page = Page(self.xml_file)
faksimile_page = FaksimilePage.get_faksimile_pages(self.faksimile_file, page_number=page.number)[0]
flc = compare_faksimile_words_line_wise.FaksimileLineComposer(faksimile_page.word_positions)
compare_faksimile_words_line_wise.mark_unmergeable_words_and_faksimile_positions(page.words, faksimile_page.word_positions)
words_on_line = [ word for word in page.words if word.line_number == 1 ]
matched_line = flc.get_line(words_on_line, 0)
self.assertTrue(words_on_line[0].text in [ fp.text for fp in matched_line ])
words_on_line[0].joined = True
matched_line[0].joined = True
words_on_line = [ word for word in page.words if word.line_number == 2 ]
matched_line = flc.get_line(words_on_line, 1)
for word in words_on_line:
self.assertTrue(word.text in [ fp.text for fp in matched_line ])
words_on_line = [ word for word in page.words if word.line_number == 42 ]
matched_line = flc.get_line(words_on_line, 22)
self.assertTrue(len(matched_line) > 0)
def test_merge_line(self):
page = Page(self.xml_file)
faksimile_page = FaksimilePage.get_faksimile_pages(self.faksimile_file, page_number=page.number)[0]
flc = compare_faksimile_words_line_wise.FaksimileLineComposer(faksimile_page.word_positions)
compare_faksimile_words_line_wise.mark_unmergeable_words_and_faksimile_positions(page.words, faksimile_page.word_positions)
words_on_line = [ word for word in page.words if word.line_number == 42 ]
new_words = []
flc.merge_lines(words_on_line, new_words, 22)
for word in new_words:
self.assertTrue(len(word.faksimile_positions) > 0)
self.assertEqual(len([word for word in words_on_line if not word.joined ]), 0)
def test_files(self):
pass
#page = Page('xml/W_II_1_page141.xml')
#faksimile_page = FaksimilePage.get_faksimile_pages('/home/knister0/ownCloud/nietzscheDE/Bearbeitung_Faksimile/Eric/Kontrolle_und_Beschriftung_der_Wortrahmen/Fertig/W-II-1,141et142.svg', page_number=page.number)[0]
#exit_code = compare_faksimile_words_line_wise.merge_faksimile_positions_and_words(page, faksimile_page.word_positions, interactive=True)
#if exit_code > 0:
# exit_code = compare_faksimile_words_line_wise.merge_faksimile_positions_and_words(page, faksimile_page.word_positions, interactive=True)
#self.assertEqual(exit_code, 0)
@unittest.skip('')
def test_manual_merge(self):
unmerged_words = [ Word(id=i, text=str(i)) for i in [ 1, 2, 3 ]]
unmerged_fps = [ FaksimilePosition(id='rect'+str(i), text=str(i)) for i in [ 1, 2, 3, 4, 5 ]]
for c in unmerged_words+unmerged_fps:
c.joined = False
word = Word(id=6, text='test', line_number=1)
fp = FaksimilePosition(id='rect222')
word.faksimile_positions.append(fp)
new_words = [ word ]
manual_shell = interactive_merger.ManualMergerShell(unmerged_words, unmerged_fps, new_words)
manual_shell.run()
def test_faksimile_line_composer(self):
faksimile_page = FaksimilePage.get_faksimile_pages(self.faksimile_file)[0]
flc = compare_faksimile_words_line_wise.FaksimileLineComposer(faksimile_page.word_positions, threshold=10)
self.assertEqual(len(flc.lines_of_faksimile_positions), 23)
flc = compare_faksimile_words_line_wise.FaksimileLineComposer(faksimile_page.word_positions, threshold=20, num_lines_with_words=23)
self.assertEqual(len(flc.lines_of_faksimile_positions), 23)
counter = 200
while counter > 0:
counter -= 1
self.assertTrue(flc.get_next_faksimile() is not None)
if __name__ == "__main__":
unittest.main()
c4science · Help