def test_read_huge_xml_file_to_dict(self):
"""Can read a very big xml file easily to return iterator of dicts when to_dict is set to True"""
if os.path.isfile(self.huge_mock_file_path):
for element in read_xml_file(self.small_mock_file_path,
records_tag=['staff'],
to_dict=True):
self.assertIsInstance(element, dict)
def test_read_xml_file_for_staff(self):
"""Converts the XML file into a Element of staff's sub elements when records_tag=staff"""
element = None
for element, tag in read_xml_file(self.small_mock_file_path,
records_tag=['staff']):
if tag == 'staff':
break
expected_element = self.expected_small_xml_output.findall('staff')[0]
self.assertIsInstance(element, Element)
self.assertEqual(ElementTree.tostring(element),
ElementTree.tostring(expected_element))
def test_read_xml_file_to_dict_for_staff(self):
"""Converts the XML string into a dict of staff's sub elements when to_dict=True, records_tag=staff"""
staff_output = {}
staff_tag = None
for element, tag in read_xml_file(self.small_mock_file_path,
records_tag=['staff'],
to_dict=True):
if tag == 'staff':
staff_output = element
staff_tag = tag
self.assertDictEqual(staff_output, self.expected_small_mock_output)
self.assertEqual(staff_tag, 'staff')
def test_read_xml_file_to_dict_for_employees(self):
"""
Converts the XML string into a list of employee's sub elements when to_dict=True, records_tag=[employees]
"""
employees_output = []
for element, tag in read_xml_file(self.small_mock_file_path,
records_tag=['employees'],
to_dict=True):
employees_output.append(element['bio'])
self.assertListEqual(
employees_output,
self.expected_small_mock_output['operations_department']
['employees'])
def test_read_xml_file_for_operations_department(self):
"""
Converts the XML file into a Element of operations department's
sub elements when records_tag=[operations_department]
"""
element = None
for element, tag in read_xml_file(self.small_mock_file_path,
records_tag=[
'operations_department'
]):
if tag == 'operations_department':
break
expected_element = self.expected_small_xml_output.findall(
'.//operations_department')[0]
self.assertIsInstance(element, Element)
self.assertEqual(ElementTree.tostring(element),
ElementTree.tostring(expected_element))
def test_read_xml_file_to_dict_for_operations_department(self):
"""
Converts the XML string into a dict of operations_department's
sub elements when to_dict=True, records_tag=[operations_department]
"""
operations_department_output = {}
operations_department_tag = None
for element, tag in read_xml_file(
self.small_mock_file_path,
records_tag=['operations_department'],
to_dict=True):
if tag == 'operations_department':
operations_department_output = element
operations_department_tag = tag
self.assertDictEqual(
operations_department_output,
self.expected_small_mock_output['operations_department'])
self.assertEqual(operations_department_tag, 'operations_department')
def test_read_xml_file_for_all_tags(self):
"""
Converts the XML file into a Element of employees'
sub elements when records_tag=[employees]
"""
for index, results in enumerate(
read_xml_file(self.small_mock_file_path,
records_tag=[
'staff', 'operations_department', 'employees'
])):
element, tag = results
if tag == 'operations_department':
self.assertIsInstance(element, Element)
elif tag == 'staff':
self.assertIsInstance(element, Element)
elif tag == 'employees':
employees_element = self.expected_small_xml_output.findall(
'.//employees')[index]
self.assertIsInstance(element, Element)
self.assertEqual(ElementTree.tostring(element),
ElementTree.tostring(employees_element))
def test_read_huge_xml_file_to_xml_element(self):
"""Can read a very big xml file easily to return iterator of xml.etree.ElementTree.Element by default"""
if os.path.isfile(self.huge_mock_file_path):
for element in read_xml_file(self.small_mock_file_path,
records_tag=['staff']):
self.assertIsInstance(element, Element)