def testIndividualIdsWork(self):
gedcomfile = gedcom.GedcomFile()
element1 = gedcom.Individual()
element2 = gedcom.Individual()
self.assertEqual(element1.id, None)
self.assertEqual(element2.id, None)
gedcomfile.add_element(element1)
gedcomfile.add_element(element2)
self.assertEqual(element1.id, '@I1@')
self.assertEqual(element2.id, '@I2@')
def return_individual_gedcom(self):
'''
This function will return the gedcom type individual
'''
individual = gedcom.Individual()
if "gender" in self.gen_data.keys():
sex = gedcom.Element(tag="SEX", value=self.getGender())
individual.add_child_element(sex)
name = gedcom.Element(tag="NAME",
value=(self.getName() + "/" + self.getSurname() +
"/"))
name_given = gedcom.Element(tag="GIVN", value=self.getName())
name_surname = gedcom.Element(tag="SURN", value=self.getSurname())
nick = ""
for nickname in self.gen_data.get("nicknames", None):
if nick != "": nick += ","
nick += nickname
name.add_child_element(name_given)
name.add_child_element(name_surname)
if (nick != ""):
name_nickname = gedcom.Element(tag="NICK", value=nick)
name.add_child_element(name_nickname)
individual.add_child_element(name)
#Review all dates to include
for date_key in EQUIVALENCE.keys():
include_date, ged_event = self.get_event_element(date_key)
if include_date: individual.add_child_element(ged_event)
return individual
def testIdAssismentIsRobust(self):
gedcomfile = gedcom.parse_string(
"0 HEAD\n0 @I1@ INDI\n1 NAME\n2 GIVN Bob\n2 SURN Cox\n\n0 TRLR")
element1 = gedcom.Individual()
self.assertEqual(element1.id, None)
gedcomfile.add_element(element1)
self.assertEqual(element1.id, '@I2@')
def test_creating_gedcom_profile(self):
'''
Test Creating a GedCom Profile
'''
name_data = "MyName"
surname = "MySurname"
total_name = name_data + "/" + surname + "/"
individual = gedcom.Individual()
name = gedcom.Element(tag="NAME", value=total_name)
name_given = gedcom.Element(tag="GIVN", value=name_data)
name_surname = gedcom.Element(tag="SURN", value=surname)
name.add_child_element(name_given)
name.add_child_element(name_surname)
individual.add_child_element(name)
profile = gedcom_profile(individual)
assert (profile.gen_data["name"] == name_data)
assert (profile.gen_data["surname"] == surname)
def testErrorWithBadLevel(self):
individual = gedcom.Individual(level='foo')
self.assertRaises(Exception, individual.set_levels_downward)
def testCanAddIndividualObj(self):
gedcomfile = gedcom.GedcomFile()
element = gedcom.Individual()
gedcomfile.add_element(element)