def test_non_existing_element(self): el = etree.fromstring(''' <root></root> ''') with self.assertRaises(IndexError): get_agent(el)
def test_multiple_agents(self): el = etree.fromstring(''' <root> <agent ROLE="foo"> <name>foo_name</name> <note>foo_note</note> </agent> <agent ROLE="bar"> <name>bar_name</name> <note>bar_note</note> </agent> </root> ''') self.assertEqual(get_agent(el, ROLE='foo'), {'name': 'foo_name', 'notes': ['foo_note']}) self.assertEqual(get_agent(el, ROLE='bar'), {'name': 'bar_name', 'notes': ['bar_note']})
def test_existing_othertype(self): el = etree.fromstring(''' <root> <agent OTHERTYPE="foo"> <name>foo_name</name> <note>foo_note</note> </agent> </root> ''') self.assertEqual(get_agent(el, OTHERTYPE='foo'), {'name': 'foo_name', 'notes': ['foo_note']})
def test_non_existing_element(self): el = etree.fromstring(''' <root></root> ''') self.assertIsNone(get_agent(el))