test_super_page.py
No OneTemporary
Actions

Subscribers

None

File Metadata

	import unittest
	from os import sep, path
	from os.path import isdir, dirname, basename
	import lxml.etree as ET
	import sys
	import sys

	sys.path.append('svgscripts')

	from datatypes.super_page import SuperPage

	class TestSuperPage(unittest.TestCase):
	def test_init(self):
	sp = SuperPage(title='test', page_number=1)
	self.assertEqual(sp.page_tree.getroot().get('title'), 'test')
	self.assertEqual(sp.page_tree.getroot().get('page-number'), '1')
	self.assertEqual(sp.tag, 'super-page')

	def test_remove_tags(self):
	sp = SuperPage(title='test', page_number=1)
	tag = 'test'
	numTags = 3
	for i in range(numTags):
	ET.SubElement(sp.page_tree.getroot(), tag)
	self.assertEqual(len(sp.page_tree.xpath('//' + tag)), numTags)
	sp.remove_tags_from_page_tree([tag])
	self.assertEqual(len(sp.page_tree.xpath('//' + tag)), 0)

	if __name__ == "__main__":
	unittest.main()