Page MenuHomec4science
Files F103590612

test_lineNumber.py
No OneTemporary
Actions

File Metadata

Created
Mon, Mar 3, 05:18

test_lineNumber.py
View Options

import unittest
from os import sep, path
from os.path import isdir, dirname
import lxml.etree as ET
import sys
dir_changed = False
if not isdir('datatypes'):
sys.path.append(dirname(sys.path[0]))
dir_changed = True
from datatypes.lineNumber import LineNumber
from datatypes.transkriptionField import TranskriptionField
class TestLineNumber(unittest.TestCase):
def setUp(self):
DATADIR = dirname(__file__) + sep + 'test_data'
if not isdir(DATADIR):
DATADIR = dirname(dirname(__file__)) + sep + 'test_data'
self.test_target_file = DATADIR + sep + 'test.xml'
self.test_source_file = DATADIR + sep + 'test_ai.svg'
self.id = 24
def test_init(self):
lnr = LineNumber(id=self.id)
self.assertEqual(lnr.id, self.id)
svg_tree = ET.parse(self.test_source_file)
tf = TranskriptionField(self.test_source_file)
raw_text_node = svg_tree.getroot().xpath('.//text[@transform="%s"]' % 'matrix(1 0 0 1 180.8755 386.7129)')[0]
lnr = LineNumber(raw_text_node=raw_text_node, transkription_field=tf)
self.assertEqual(lnr.id, self.id)
self.assertEqual(lnr.bottom, 311.8129)
tree = ET.ElementTree(ET.Element('svg'))
node = ET.SubElement(tree.getroot(), 'text', attrib={'transform': 'matrix(1 0 0 1 180.8755 386.7129)'})
node.text = str(self.id)
lnr = LineNumber(raw_text_node=node, transkription_field=tf)
self.assertEqual(lnr.id, self.id)
self.assertEqual(lnr.bottom, 311.8129)
node = ET.SubElement(tree.getroot(), 'text', attrib={'transform': 'matrix(1 0 0 1 180.8755 386.7129)'})
subnode = ET.SubElement(node, 'tspan')
subnode.text = '1'
subnode = ET.SubElement(node, 'tspan')
subnode.text = '0'
lnr = LineNumber(raw_text_node=node, transkription_field=tf)
self.assertEqual(lnr.id, 10)
self.assertEqual(lnr.bottom, 311.8129)
def test_init_from_xml(self):
xml_tree = ET.parse(self.test_target_file)
line_numbers = [ LineNumber(xml_text_node=node) for node in xml_tree.getroot().xpath('.//line-number') ]
self.assertEqual(len(line_numbers), 49)
self.assertEqual(line_numbers[0].id, 1)
self.assertEqual(line_numbers[48].id, 49)
def test_attach_object_to_tree(self):
empty_tree = ET.ElementTree(ET.Element('page'))
lnr = LineNumber(id=self.id)
lnr.attach_object_to_tree(empty_tree)
lnr_nodes = empty_tree.getroot().xpath('//' + LineNumber.XML_TAG() + '[@id="%s"]' % self.id)
self.assertEqual(len(lnr_nodes), 1)
self.assertEqual(lnr_nodes[0].get('id'), str(self.id))
def test_IS_A_LINE_NUMBER(self):
tree = ET.ElementTree(ET.Element('svg'))
node = ET.SubElement(tree.getroot(), 'text')
node.text = '2'
self.assertEqual(LineNumber.IS_A_LINE_NUMBER(node), True)
node = ET.SubElement(tree.getroot(), 'text')
subnode = ET.SubElement(node, 'tspan')
subnode.text = '1'
subnode = ET.SubElement(node, 'tspan')
subnode.text = '0'
self.assertEqual(LineNumber.IS_A_LINE_NUMBER(node), True)
subnode.text = 'x'
self.assertEqual(LineNumber.IS_A_LINE_NUMBER(node), False)
def test_get_semanticAndDataDict(self):
xml_tree = ET.parse(self.test_target_file)
line_number = [ LineNumber(xml_text_node=node) for node in xml_tree.getroot().xpath('.//line-number') ][1]
#self.assertEqual(line_number.get_data_dictionary()['body'].get('id'), 2)
#print(line_number.get_semantic_dictionary())
if __name__ == "__main__":
unittest.main()

Event Timeline

c4science · Help