def test_read_xml_string_for_staff(self):
"""Converts the XML string into a Element of staff's sub elements when records_tag=staff"""
element = None
for element, tag in read_xml_string(self.xml_string,
records_tag=['staff']):
if tag == 'staff':
break
expected_element = self.expected_xml_output.findall('.//staff')[0]
self.assertIsInstance(element, Element)
self.assertEqual(ElementTree.tostring(element),
ElementTree.tostring(expected_element))
def test_read_xml_string_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_string(self.xml_string,
records_tag=['staff'],
to_dict=True):
staff_output = element
staff_tag = tag
self.assertDictEqual(staff_output, self.expected_output)
self.assertEqual(staff_tag, 'staff')
def test_read_xml_string_for_operations_department(self):
"""
Converts the XML string into a Element of operations department's
sub elements when records_tag=operations_department
"""
element = None
for element, tag in read_xml_string(
self.xml_string, records_tag=['operations_department']):
if tag == 'operations_department':
break
expected_element = self.expected_xml_output.findall(
'.//operations_department')[0]
self.assertIsInstance(element, Element)
self.assertEqual(ElementTree.tostring(element),
ElementTree.tostring(expected_element))
def test_read_xml_string_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_string(self.xml_string,
records_tag=['employees'],
to_dict=True):
if tag == "employees":
employees_output.append(element['bio'])
self.assertListEqual(
employees_output,
self.expected_output['operations_department']['employees'])
def test_read_xml_string_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_string(
self.xml_string,
records_tag=['operations_department'],
to_dict=True):
operations_department_output = element
operations_department_tag = tag
self.assertDictEqual(operations_department_output,
self.expected_output['operations_department'])
self.assertEqual(operations_department_tag, 'operations_department')
def test_read_xml_string_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_string(self.xml_string,
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_xml_output.findall(
'.//employees')[index]
self.assertIsInstance(element, Element)
self.assertEqual(ElementTree.tostring(element),
ElementTree.tostring(employees_element))