def test_xml_etree2xml_1(self):
dude = People(idref="PER_JAMESK", name="James Kunze", kind="Person who cleans up all your broken tests")
dude.save()
et = xml_mods2etree()
xml_text = xml_etree2xml(et).decode("ascii")
self.assertTrue(type(xml_text) is UnicodeType)
obama = People(idref="PER_BROBMA", name="Barack Obama", kind="President")
obama.save()
et = xml_mods2etree()
xml_text = xml_etree2xml(et).decode("ascii")
self.assertTrue(type(xml_text) is UnicodeType)
def test_xml_merge_entitycount(self):
Crises.objects.all().delete()
People.objects.all().delete()
Organizations.objects.all().delete()
List_Item.objects.all().delete()
xml_file = open("wcdb/WCDB2.xml", "r")
xml_file_tomerge = open("wcdb/crisix.xml", "r")
xsd = open("wcdb/WCDB2.xsd.xml", "r")
et = xml_reader(xml_file, xsd)
xml_etree2mods(et.getroot())
incoming_data = xml_reader(xml_file_tomerge, xsd)
xml_etree2mods(incoming_data.getroot())
# now is when you want to know how many videos there are in the database
root = xml_mods2etree().getroot()
crisis_cnt = 0
person_cnt = 0
org_cnt = 0
for child in list(root):
if child.tag == "Crisis":
crisis_cnt += 1
elif child.tag == "Organization":
org_cnt += 1
elif child.tag == "Person":
person_cnt += 1
self.assertEqual(crisis_cnt, 19)
self.assertEqual(person_cnt, 19)
self.assertEqual(org_cnt, 20)
def test_xml_mods2etree_1(self):
et = xml_mods2etree()
root = et.getroot()
self.assertEqual(type(et), type(ET.ElementTree("")))
self.assertEqual(root.tag, "WorldCrises")
for child in list(root):
valid = child.tag == "Crisis" or child.tag == "Organization" or child.tag == "Person"
self.assertEqual(valid, True)
def test_xml_mods2etree_2(self):
et = xml_mods2etree()
root = et.getroot()
self.assertEqual(type(et), type(ET.ElementTree("")))
self.assertEqual(root.tag, "WorldCrises")
crisis_cnt = 0
person_cnt = 0
org_cnt = 0
for child in list(root):
if child.tag == "Crisis":
crisis_cnt += 1
elif child.tag == "Organization":
org_cnt += 1
elif child.tag == "Person":
person_cnt += 1
self.assertEqual(crisis_cnt, 0)
self.assertEqual(person_cnt, 0)
self.assertEqual(org_cnt, 0)
def test_xml_mods2etree_3(self):
p = People(idref="PER_TCANNO", name="Taylor Cannon", kind="badass")
p.save()
et = xml_mods2etree()
root = et.getroot()
self.assertEqual(type(et), type(ET.ElementTree("")))
self.assertEqual(root.tag, "WorldCrises")
crisis_cnt = 0
person_cnt = 0
org_cnt = 0
for child in list(root):
if child.tag == "Crisis":
crisis_cnt += 1
elif child.tag == "Organization":
org_cnt += 1
elif child.tag == "Person":
person_cnt += 1
self.assertEqual(crisis_cnt, 0)
self.assertEqual(person_cnt, 1)
self.assertEqual(org_cnt, 0)
def test_xml_etree2xml_1(self) : et = xml_mods2etree() xml_text = xml_etree2xml(et) self.assertEqual(type(xml_text) is UnicodeType, True)
