def test3(self): male1 = Individual("P01") female1 = Individual("P02") child1 = Individual("P03") family1 = Family("F01") family1.add_child(child1) male1.set_deathDate(["5", "MAR", "2000"]) female1.set_deathDate(["9", "APR", "2002"]) child1.set_birthDate(["6", "MAR", "2000"]) family1.set_husband(male1) family1.set_wife(female1) self.assertTrue(family1.birth_before_death_of_parents())
def test_date_compare(self): fam_1 = Family("01") ind_1 = Individual("01") ind_1.set_birthDate((2019, 9, 1)) ind_1.set_parentFamily(fam_1) fam_1.set_marriedDate((2019, 9, 2)) ind_1.set_deathDate((2020, 9, 1)) fam_2 = Family("02") ind_2 = Individual("02") ind_2.set_birthDate((2019, 8, 1)) ind_2.set_parentFamily((fam_2)) fam_2.set_marriedDate((2019, 7, 1)) ind_2.set_deathDate((2018, 8, 1)) assert ind_1.birth_before_marriage() == True assert ind_2.birth_before_marriage() == False assert ind_1.birth_before_death() == True assert ind_2.birth_before_death() == False
def test_US08_birth_before_marriage_of_parents(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") child1 = Individual("P03") t1.add_child(child1) t1.set_marriedDate(['8', 'SEP', '2000']) child1.set_birthDate(["6", "JAN", "1998"]) t1.set_husband(male1) t1.set_wife(female1) # --------------------------------- t2 = Family("F02") male2 = Individual("P04") female2 = Individual("P05") child2 = Individual("P06") t2.add_child(child2) t2.set_marriedDate(['8', 'SEP', '2000']) child2.set_birthDate(["6", "JAN", "2001"]) t2.set_husband(male2) t2.set_wife(female2) # --------------------------------- t3 = Family("F03") male3 = Individual("P07") female3 = Individual("P08") child3 = Individual("P09") t3.add_child(child3) t3.set_marriedDate(['6', 'MAR', '2000']) child3.set_birthDate(["6", "MAR", "2000"]) t3.set_husband(male3) t3.set_wife(female3) # --------------------------------- assert t1.birth_before_marriage_of_parents() == False assert t2.birth_before_marriage_of_parents() == True assert t3.birth_before_marriage_of_parents() == False
def test_US09_birth_before_death_of_parent(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") child1 = Individual("P03") t1.add_child(child1) male1.set_deathDate(["5", "MAR", "2000"]) female1.set_deathDate(["9", "APR", "2002"]) child1.set_birthDate(["6", "JAN", "1998"]) t1.set_husband(male1) t1.set_wife(female1) # --------------------------------- t2 = Family("F02") male2 = Individual("P04") female2 = Individual("P05") child2 = Individual("P06") t2.add_child(child2) male2.set_deathDate(["5", "MAR", "2000"]) female2.set_deathDate(["9", "APR", "2002"]) child2.set_birthDate(["6", "JAN", "2001"]) t2.set_husband(male2) t2.set_wife(female2) # --------------------------------- t3 = Family("F03") male3 = Individual("P07") female3 = Individual("P08") child3 = Individual("P09") t3.add_child(child3) male3.set_deathDate(["5", "MAR", "2000"]) female3.set_deathDate(["9", "APR", "2002"]) child3.set_birthDate(["6", "MAR", "2000"]) t3.set_husband(male3) t3.set_wife(female3) # --------------------------------- assert t1.birth_before_death_of_parents() == True assert t2.birth_before_death_of_parents() == False assert t3.birth_before_death_of_parents() == True
class TestSprint1(unittest.TestCase): def setUp(self): SUPPORT_TAGS = { "INDI", "NAME", "SEX", "BIRT", "DEAT", "FAMC", "FAMS", "FAM", "MARR", "HUSB", "WIFE", "CHIL", "DIV", "DATE", "HEAD", "TRLR", "NOTE" } # self.G1 = Gedcom('../testing_files/right.ged', SUPPORT_TAGS) # self.G2 = Gedcom('../testing_files/wrong.ged', SUPPORT_TAGS) self.ind_1 = Individual("01") self.ind_2 = Individual("02") self.ind_3 = Individual("03") self.fam_1 = Family("01") self.fam_2 = Family("02") def tearDown(self): self.ind_1 = Individual("01") self.ind_2 = Individual("02") self.ind_3 = Individual("03") self.fam_1 = Family("01") self.fam_2 = Family("02") def test_US11_no_bigamy(self): self.ind_1.set_birthDate(["09", "APR", "1997"]) self.ind_2.set_birthDate(["19", "DEC", "1997"]) self.ind_1.add_to_family(self.fam_1) self.fam_1.set_marriedDate(["01", "JUN", "2017"]) self.assertTrue(self.ind_1.no_bigamy()) self.fam_2.set_marriedDate(["05", "JUN", "2016"]) self.ind_1.add_to_family(self.fam_2) self.assertFalse(self.ind_1.no_bigamy()) self.fam_2.set_divorcedDate(("01", "JAN", "2017")) self.assertTrue(self.ind_1.no_bigamy()) self.fam_2.set_divorcedDate(("01", "AUG", "2017")) self.assertFalse(self.ind_1.no_bigamy()) self.fam_1.set_divorcedDate(("01", "DEC", "2018")) self.assertFalse(self.ind_1.no_bigamy()) self.fam_2.set_divorcedDate(("01", "JAN", "2017")) self.assertTrue(self.ind_1.no_bigamy()) def test_US02_birth_before_marriage(self): self.ind_1.set_birthDate(["09", "APR", "1997"]) self.ind_1.add_to_family(self.fam_1) self.fam_1.set_marriedDate(["01", "JUN", "2018"]) self.assertTrue(self.ind_1.birth_before_marriage()) def test_US03_birth_before_death(self): self.ind_1.set_birthDate(["09", "APR", "1997"]) self.ind_1.set_deathDate(["01", "JUN", "2018"]) self.assertTrue(self.ind_1.birth_before_death()) def test_US07_less_then_150_years_old(self): self.ind_1.set_birthDate(["09", "APR", "1997"]) self.assertTrue(self.ind_1.less_then_150_years_old()) self.ind_2.set_birthDate(["09", "APR", "997"]) self.assertFalse(self.ind_2.less_then_150_years_old()) def test_US04_marriage_before_divorce(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") male1.set_deathDate(['8', 'SEP', '2010']) female1.set_deathDate(['8', 'SEP', '2011']) t1.set_husband(male1) t1.set_wife(female1) t1.set_marriedDate(['8', 'SEP', '2000']) t1.set_divorcedDate(['8', 'SEP', '2009']) # --------------------------------- t2 = Family("F02") male2 = Individual("P03") female2 = Individual("P04") male2.set_deathDate(['8', 'SEP', '2012']) female2.set_deathDate(['8', 'SEP', '2013']) t2.set_husband(male2) t2.set_wife(female2) t2.set_marriedDate(['8', 'SEP', '2005']) t2.set_divorcedDate(['8', 'SEP', '2004']) # --------------------------------- assert t1.marriage_before_divorce() == True assert t2.marriage_before_divorce() == False def test_US05_marriage_before_death(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") male1.set_deathDate(['8', 'SEP', '2010']) female1.set_deathDate(['8', 'SEP', '2011']) t1.set_husband(male1) t1.set_wife(female1) t1.set_marriedDate(['8', 'SEP', '2000']) # --------------------------------- t2 = Family("F02") male2 = Individual("P03") female2 = Individual("P04") male2.set_deathDate(['8', 'SEP', '1999']) female2.set_deathDate(['9', 'SEP', '2011']) t2.set_husband(male2) t2.set_wife(female2) t2.set_marriedDate(['8', 'SEP', '2000']) # --------------------------------- t3 = Family("F03") male3 = Individual("P05") female3 = Individual("P06") male3.set_deathDate(['8', 'SEP', '2003']) female3.set_deathDate(['9', 'SEP', '1998']) t3.set_husband(male3) t3.set_wife(female3) t3.set_marriedDate(['8', 'SEP', '2000']) # --------------------------------- t4 = Family("F04") male4 = Individual("P07") female4 = Individual("P08") male4.set_deathDate(['8', 'SEP', '1998']) female4.set_deathDate(['9', 'SEP', '1999']) t4.set_husband(male4) t4.set_wife(female4) t4.set_marriedDate(['8', 'SEP', '2000']) # --------------------------------- t5 = Family("F05") male5 = Individual("P09") female5 = Individual("P10") male5.set_deathDate(['8', 'SEP', '2009']) female5.set_deathDate(['8', 'SEP', '2009']) t5.set_husband(male5) t5.set_wife(female5) t5.set_marriedDate(['8', 'SEP', '2009']) # --------------------------------- assert t1.marriage_before_death() == True assert t2.marriage_before_death() == False assert t3.marriage_before_death() == False assert t4.marriage_before_death() == False assert t5.marriage_before_death() == True def test_US06_divorse_before_death(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") male1.set_deathDate(["5", "MAR", "2000"]) female1.set_deathDate(["9", "APR", "2002"]) t1.set_husband(male1) t1.set_wife(female1) t1.set_divorcedDate(["1", "JAN", "1999"]) # --------------------------------- t2 = Family("F02") male2 = Individual("P03") female2 = Individual("P04") male2.set_deathDate(["5", "MAR", "2000"]) female2.set_deathDate(["9", "APR", "2002"]) t2.set_husband(male2) t2.set_wife(female2) t2.set_divorcedDate(["1", "JAN", "2003"]) # --------------------------------- assert t1.divorce_before_death() == True assert t2.divorce_before_death() == False def test_US08_birth_before_marriage_of_parents(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") child1 = Individual("P03") t1.add_child(child1) t1.set_marriedDate(['8', 'SEP', '2000']) child1.set_birthDate(["6", "JAN", "1998"]) t1.set_husband(male1) t1.set_wife(female1) # --------------------------------- t2 = Family("F02") male2 = Individual("P04") female2 = Individual("P05") child2 = Individual("P06") t2.add_child(child2) t2.set_marriedDate(['8', 'SEP', '2000']) child2.set_birthDate(["6", "JAN", "2001"]) t2.set_husband(male2) t2.set_wife(female2) # --------------------------------- t3 = Family("F03") male3 = Individual("P07") female3 = Individual("P08") child3 = Individual("P09") t3.add_child(child3) t3.set_marriedDate(['6', 'MAR', '2000']) child3.set_birthDate(["6", "MAR", "2000"]) t3.set_husband(male3) t3.set_wife(female3) # --------------------------------- assert t1.birth_before_marriage_of_parents() == False assert t2.birth_before_marriage_of_parents() == True assert t3.birth_before_marriage_of_parents() == False def test_US09_birth_before_death_of_parent(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") child1 = Individual("P03") t1.add_child(child1) male1.set_deathDate(["5", "MAR", "2000"]) female1.set_deathDate(["9", "APR", "2002"]) child1.set_birthDate(["6", "JAN", "1998"]) t1.set_husband(male1) t1.set_wife(female1) # --------------------------------- t2 = Family("F02") male2 = Individual("P04") female2 = Individual("P05") child2 = Individual("P06") t2.add_child(child2) male2.set_deathDate(["5", "MAR", "2000"]) female2.set_deathDate(["9", "APR", "2002"]) child2.set_birthDate(["6", "JAN", "2001"]) t2.set_husband(male2) t2.set_wife(female2) # --------------------------------- t3 = Family("F03") male3 = Individual("P07") female3 = Individual("P08") child3 = Individual("P09") t3.add_child(child3) male3.set_deathDate(["5", "MAR", "2000"]) female3.set_deathDate(["9", "APR", "2002"]) child3.set_birthDate(["6", "MAR", "2000"]) t3.set_husband(male3) t3.set_wife(female3) # --------------------------------- assert t1.birth_before_death_of_parents() == True assert t2.birth_before_death_of_parents() == False assert t3.birth_before_death_of_parents() == True def test_US10_marriage_after_14(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") male1.set_birthDate(['8', 'SEP', '2000']) female1.set_birthDate(['8', 'SEP', '2000']) t1.set_husband(male1) t1.set_wife(female1) t1.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t2 = Family("F02") male2 = Individual("P03") female2 = Individual("P04") male2.set_birthDate(['7', 'SEP', '2000']) female2.set_birthDate(['8', 'SEP', '2000']) t2.set_husband(male2) t2.set_wife(female2) t2.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t3 = Family("F03") male3 = Individual("P05") female3 = Individual("P06") male3.set_birthDate(['8', 'SEP', '2000']) female3.set_birthDate(['7', 'SEP', '2000']) t3.set_husband(male3) t3.set_wife(female3) t3.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t4 = Family("F04") male4 = Individual("P07") female4 = Individual("P08") male4.set_birthDate(['1', 'SEP', '1990']) female4.set_birthDate(['2', 'SEP', '1990']) t4.set_husband(male4) t4.set_wife(female4) t4.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t5 = Family("F05") male5 = Individual("P09") female5 = Individual("P10") male5.set_birthDate(['09', 'APR', '1997']) female5.set_birthDate(['19', 'DEC', '1997']) t5.set_husband(male5) t5.set_wife(female5) t5.set_marriedDate(['1', 'JUN', '2007']) # -------------------------------------------------- assert t1.marriage_after_14() == False assert t2.marriage_after_14() == False assert t3.marriage_after_14() == False assert t4.marriage_after_14() == True assert t5.marriage_after_14() == False def test_US13_siblings_spacing(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") t5 = Family("t5") t6 = Family("t6") p1 = Individual("p1") p1.set_birthDate(("1", "JAN", "1990")) p2 = Individual("p2") p2.set_birthDate(("1", "JAN", "1990")) p3 = Individual("p3") p3.set_birthDate(("1", "SEP", "1990")) p4 = Individual("p4") p4.set_birthDate(("2", "JAN", "1990")) p5 = Individual("p5") p5.set_birthDate(("3", "JAN", "1990")) p6 = Individual("p6") p6.set_birthDate(("30", "MAY", "1990")) # -------------------------------------------------- t1.add_child(p1) t1.add_child(p2) t2.add_child(p1) t2.add_child(p3) t3.add_child(p1) t3.add_child(p4) t4.add_child(p1) t4.add_child(p5) t5.add_child(p1) t5.add_child(p6) t6.add_child(p1) t6.add_child(p3) t6.add_child(p6) # -------------------------------------------------- assert t1.siblings_spacing() == True assert t2.siblings_spacing() == True assert t3.siblings_spacing() == True assert t4.siblings_spacing() == False assert t5.siblings_spacing() == False assert t6.siblings_spacing() == False def test_US14_multiple_births_lessOrEqual_than_5(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") p1 = Individual("p1") p1.set_birthDate(("1", "JAN", "1990")) p2 = Individual("p2") p2.set_birthDate(("1", "JAN", "1990")) p3 = Individual("p3") p3.set_birthDate(("1", "JAN", "1990")) p4 = Individual("p4") p4.set_birthDate(("3", "JAN", "1990")) p5 = Individual("p5") p5.set_birthDate(("2", "JAN", "1990")) p6 = Individual("p6") p6.set_birthDate(("30", "MAY", "1990")) p7 = Individual("p7") p7.set_birthDate(("2", "JAN", "1990")) p8 = Individual("p8") p8.set_birthDate(("2", "JAN", "1990")) p9 = Individual("p9") p9.set_birthDate(("2", "SEP", "1990")) p10 = Individual("p10") p10.set_birthDate(("2", "SEP", "1990")) p11 = Individual("p11") p11.set_birthDate(("3", "SEP", "1990")) p12 = Individual("p12") p12.set_birthDate(("3", "SEP", "1990")) p13 = Individual("p13") p13.set_birthDate(("3", "SEP", "1990")) # ------------------------------- t1.add_child(p1) t1.add_child(p2) t1.add_child(p3) t1.add_child(p4) t1.add_child(p5) t1.add_child(p6) t1.add_child(p7) t1.add_child(p8) t2.add_child(p1) t2.add_child(p2) t2.add_child(p3) t2.add_child(p4) t2.add_child(p5) t2.add_child(p6) t3.add_child(p3) t3.add_child(p4) t3.add_child(p5) t3.add_child(p6) t3.add_child(p7) t3.add_child(p8) t3.add_child(p9) t3.add_child(p10) t3.add_child(p11) t3.add_child(p12) t3.add_child(p13) # --------------------------------- assert t1.multiple_births_lessOrEqual_than_5() == False assert t2.multiple_births_lessOrEqual_than_5() == True assert t3.multiple_births_lessOrEqual_than_5() == False def test_US12_parents_not_too_old(self): t1 = Family("t1") t2 = Family("t2") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") p5 = Individual("p5") p6 = Individual("p6") t1.set_wife(p1) t1.set_husband(p2) t1.add_child(p3) t2.set_wife(p4) t2.set_husband(p5) t2.add_child(p6) p1.set_birthDate(["1", "JAN", "1990"]) p2.set_birthDate(["1", "JAN", "1990"]) p4.set_birthDate(["1", "JAN", "1790"]) p5.set_birthDate(["1", "JAN", "1790"]) p3.set_birthDate(["1", "JAN", "2010"]) p6.set_birthDate(["1", "JAN", "2000"]) # --------------------------------- assert t1.parents_not_too_old() == True assert t2.parents_not_too_old() == False def test_US15_Fewer_than_15_siblings(self): t1 = Family("t1") t2 = Family("t2") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") p5 = Individual("p5") p6 = Individual("p6") p7 = Individual("p7") p8 = Individual("p8") p9 = Individual("p9") p10 = Individual("p10") p11 = Individual("p11") p12 = Individual("p12") p13 = Individual("p13") p14 = Individual("p14") p15 = Individual("p15") p16 = Individual("p16") p17 = Individual("p17") p18 = Individual("p18") p19 = Individual("p19") p20 = Individual("p20") p21 = Individual("p21") p22 = Individual("p22") p23 = Individual("p23") p24 = Individual("p24") p25 = Individual("p25") p26 = Individual("p26") p27 = Individual("p27") p28 = Individual("p28") # --------------------------------- t1.add_child(p1) t1.add_child(p2) t1.add_child(p3) t1.add_child(p4) t1.add_child(p5) t1.add_child(p6) t1.add_child(p7) t1.add_child(p8) t1.add_child(p9) t1.add_child(p10) t1.add_child(p11) t1.add_child(p12) t2.add_child(p13) t2.add_child(p14) t2.add_child(p15) t2.add_child(p16) t2.add_child(p17) t2.add_child(p18) t2.add_child(p19) t2.add_child(p20) t2.add_child(p21) t2.add_child(p22) t2.add_child(p23) t2.add_child(p24) t2.add_child(p25) t2.add_child(p26) t2.add_child(p27) t2.add_child(p28) # --------------------------------- assert t1.fewer_than_15_siblings() == True assert t2.fewer_than_15_siblings() == False def test_US21_Correct_Gender_For_Role(self): t1 = Family("t1") t2 = Family("t2") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") # --------------------------------- t1.set_wife(p1) t1.set_husband(p2) t2.set_wife(p3) t2.set_husband(p4) p1.set_gender('F') p2.set_gender('M') p4.set_gender('F') p3.set_gender('M') # --------------------------------- assert t1.correct_gender_for_role() == True assert t2.correct_gender_for_role() == False def test_US24_Unique_families_by_spouses(self): SUPPORT_TAGS = { "INDI", "NAME", "SEX", "BIRT", "DEAT", "FAMC", "FAMS", "FAM", "MARR", "HUSB", "WIFE", "CHIL", "DIV", "DATE", "HEAD", "TRLR", "NOTE" } self.G1 = Gedcom('testing_files/Jiashu_Wang.ged', SUPPORT_TAGS) G2 = Gedcom('testing_files/MichealFahimGEDCOM.ged', SUPPORT_TAGS) G3 = Gedcom('testing_files/mock-family.ged', SUPPORT_TAGS) # --------------------------------- assert self.G1.unique_families_by_spouses() == True assert G2.unique_families_by_spouses() == True assert G3.unique_families_by_spouses() == True def test_US25_Unique_first_names_in_families(self): SUPPORT_TAGS = { "INDI", "NAME", "SEX", "BIRT", "DEAT", "FAMC", "FAMS", "FAM", "MARR", "HUSB", "WIFE", "CHIL", "DIV", "DATE", "HEAD", "TRLR", "NOTE" } self.G1 = Gedcom('testing_files/Jiashu_Wang.ged', SUPPORT_TAGS) G2 = Gedcom('testing_files/MichealFahimGEDCOM.ged', SUPPORT_TAGS) G3 = Gedcom('testing_files/mock-family.ged', SUPPORT_TAGS) # --------------------------------- assert self.G1.unique_first_names_in_families() == True assert G2.unique_first_names_in_families() == True assert G3.unique_first_names_in_families() == True def test_US22_UniqueId(self): pass # finished in main funciton def test_US23_unique_name_and_birth_date(self): SUPPORT_TAGS = { "INDI", "NAME", "SEX", "BIRT", "DEAT", "FAMC", "FAMS", "FAM", "MARR", "HUSB", "WIFE", "CHIL", "DIV", "DATE", "HEAD", "TRLR", "NOTE" } self.G1 = Gedcom('testing_files/Jiashu_Wang.ged', SUPPORT_TAGS) G2 = Gedcom('testing_files/MichealFahimGEDCOM.ged', SUPPORT_TAGS) G3 = Gedcom('testing_files/mock-family.ged', SUPPORT_TAGS) # -------------------------------------------------- assert self.G1.unique_name_and_birth_date() == True assert G2.unique_name_and_birth_date() == True assert G3.unique_name_and_birth_date() == True def test_US18_Siblings_should_not_marry(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") t5 = Family("t5") t6 = Family("t6") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") # -------------------------------------------------- t1.set_husband(p1) t1.set_wife(p2) t4.set_husband(p3) t4.set_wife(p4) ''' t2.add_child(p1) t3.add_child(p2) t4.set_husband(p3) t4.set_wife(p4) t5.add_child(p3) t5.add_child(p4) ''' # -------------------------------------------------- p1.set_parentFamily(t2) p2.set_parentFamily(t3) p3.set_parentFamily(t5) p4.set_parentFamily(t5) # -------------------------------------------------- assert t1.siblings_should_not_marry() == True #assert t2.siblings_should_not_marry() == True assert t4.siblings_should_not_marry() == False #assert t5.siblings_should_not_marry() == False def test_US19_First_cousins_should_not_marry(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") t5 = Family("t5") t6 = Family("t6") t7 = Family("t7") t8 = Family("t8") t9 = Family("t9") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") p5 = Individual("p5") p6 = Individual("p6") p7 = Individual("p7") p8 = Individual("p8") # -------------------------------------------------- ''' t1.add_child(p1) t1.add_child(p2) t2.set_wife(p1) t2.add_child(p3) t3.set_wife(p2) t3.add_child(p4) t4.set_husband(p3) t5.set_wife(p4) t6.add_child(p5) t6.add_child(p6) t7.set_wife(p5) t8.set_wife(p6) t7.add_child(p7) t8.add_child(p8) t9.set_wife(p7) t9.set_husband(p8) ''' # -------------------------------------------------- p3.set_parentFamily(t1) t1.set_husband(p8) t1.set_wife(p7) p8.set_parentFamily(t2) p7.set_parentFamily(t3) t2.add_child(p8) t3.add_child(p7) assert p3.first_cousins_should_not_marry() == True #assert p4.first_cousins_should_not_marry()==True def test_US16_Male_last_names(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") t5 = Family("t5") t6 = Family("t6") t7 = Family("t7") t8 = Family("t8") t9 = Family("t9") t10 = Family("t10") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") p5 = Individual("p5") p6 = Individual("p6") p7 = Individual("p7") p8 = Individual("p8") p9 = Individual("p9") p10 = Individual("p10") # -------------------------------------------------- t1.set_husband(p1) t1.add_child(p2) t1.add_child(p3) t2.set_husband(p2) t3.set_husband(p3) t2.add_child(p4) t3.add_child(p5) t4.set_husband(p4) t5.set_husband(p5) t6.set_husband(p6) t6.add_child(p7) t6.add_child(p8) t7.set_husband(p7) t8.set_husband(p8) t7.add_child(p9) t8.add_child(p10) t9.set_husband(p9) t10.set_husband(p10) p1.set_gender("M") p1.set_name("Charles Glass") p2.set_gender("M") p2.set_name("Charles Glass") p3.set_gender("M") p3.set_name("Charles Glass") p4.set_gender("M") p4.set_name("Charles Glass") p5.set_gender("M") p5.set_name("Charles Glass") p6.set_gender("M") p6.set_name("Charles Glass") p7.set_gender("M") p7.set_name("Charles Glass") p8.set_gender("M") p8.set_name("Charles WDNMD") p9.set_gender("M") p9.set_name("Charles Glass") p10.set_gender("M") p10.set_name("Charles Glass") # -------------------------------------------------- assert t3.male_last_names() == True assert t8.male_last_names() == False def test_US17_No_marriages_to_descendants(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") p5 = Individual("p5") p6 = Individual("p6") p7 = Individual("p7") p8 = Individual("p8") p9 = Individual("p9") t1.set_husband(p1) t1.set_wife(p2) t1.add_child(p3) t2.set_wife(p3) t2.set_husband(p4) t2.add_child(p5) t3.set_husband(p6) t3.set_wife(p7) t3.add_child(p8) t4.set_husband(p6) t4.set_wife(p8) t4.add_child(p9) # -------------------------------------------------- #assert p3.no_marriages_to_descendants()==True #assert p6.no_marriages_to_descendants()==False #assert p8.no_marriages_to_descendants()==True def test_US27_eInclude_individual_ags(self): SUPPORT_TAGS = { "INDI", "NAME", "SEX", "BIRT", "DEAT", "FAMC", "FAMS", "FAM", "MARR", "HUSB", "WIFE", "CHIL", "DIV", "DATE", "HEAD", "TRLR", "NOTE" } self.G1 = Gedcom('testing_files/Jiashu_Wang.ged', SUPPORT_TAGS) G2 = Gedcom('testing_files/MichealFahimGEDCOM.ged', SUPPORT_TAGS) G3 = Gedcom('testing_files/mock-family.ged', SUPPORT_TAGS) # -------------------------------------------------- ''' assert self.G1.include_individual_ages() == True assert G2.include_individual_ages() == True assert G3.include_individual_ages() == True ''' def test_US28_Order_siblings_by_age(self): t1 = Family("t1") t2 = Family("t2") p1 = Individual("p1") p1.set_birthDate((1990, 4, 1)) p2 = Individual("p2") p2.set_birthDate((1990, 1, 1)) p3 = Individual("p3") p3.set_birthDate((1990, 9, 1)) p4 = Individual("p4") p4.set_birthDate((1987, 1, 1)) p5 = Individual("p5") p5.set_birthDate((2019, 1, 1)) p6 = Individual("p6") p6.set_birthDate((2017, 5, 30)) p7 = Individual("p7") p7.set_birthDate((2018, 3, 30)) p8 = Individual("p8") p8.set_birthDate((2019, 8, 30)) # -------------------------------------------------- t1.add_child(p1) t1.add_child(p2) t1.add_child(p3) t1.add_child(p4) t1.add_child(p5) t1.add_child(p6) t2.add_child(p1) t2.add_child(p2) t2.add_child(p3) t2.add_child(p4) t2.add_child(p7) t2.add_child(p8) # -------------------------------------------------- assert t1.order_siblings_by_age() == [p4, p2, p1, p3, p6, p5] assert t2.order_siblings_by_age() == [p4, p2, p1, p3, p7, p8] def test_US20_Aunts_and_uncles(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") t5 = Family("t5") t6 = Family("t6") t7 = Family("t7") t8 = Family("t8") t9 = Family("t9") t10 = Family("t10") t11 = Family("t11") t12 = Family("t12") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") p5 = Individual("p5") p6 = Individual("p6") p7 = Individual("p7") p8 = Individual("p8") p9 = Individual("p9") p10 = Individual("p10") p11 = Individual("p11") # -------------------------------------------------- p11.set_parentFamily(t1) t1.set_husband(p1) t1.set_wife(p2) p1.set_parentFamily(t2) p2.set_parentFamily(t3) #t2.set_husband(p3) #t2.set_wife(p4) #t3.set_husband(p5) #t3.set_wife(p6) t2.set_children([p1, p7, p8]) t3.set_children([p2, p9, p10]) ''' t1.add_child(p3) t1.add_child(p4) t2.set_husband(p3) t3.set_wife(p4) t2.add_child(p5) t3.add_child(p6) t4.set_husband(p7) t4.set_wife(p8) t4.add_child(p9) t4.add_child(p10) t5.set_husband(p9) t5.add_child(p11) t6.set_husband(p10) t6.set_wife(p11) ''' # -------------------------------------------------- assert p11.aunts_and_uncles() == True #assert p10.aunts_and_uncles()==False def test_US26_Corresponding_entries(self): SUPPORT_TAGS = { "INDI", "NAME", "SEX", "BIRT", "DEAT", "FAMC", "FAMS", "FAM", "MARR", "HUSB", "WIFE", "CHIL", "DIV", "DATE", "HEAD", "TRLR", "NOTE" } self.G1 = Gedcom('testing_files/Jiashu_Wang.ged', SUPPORT_TAGS) G2 = Gedcom('testing_files/MichealFahimGEDCOM.ged', SUPPORT_TAGS) G3 = Gedcom('testing_files/mock-family.ged', SUPPORT_TAGS) # -------------------------------------------------- assert self.G1.corresponding_entries() == True assert G2.corresponding_entries() == True assert G3.corresponding_entries() == True def test_US29_list_deceased(self): self.assertEqual(self.G1.listDeceased().len(), 5) self.assertNotEqual(self.G1.listDeceased().len(), 3) deceasedPeople = [] for indi in deceasedPeople: self.assertIn(indi, self.G1.listDeceased()) #List all living married people in a GEDCOM file def test_US30_list_living_married(self): self.assertEqual(self.G1.listLivingMarried().len(), 5) self.assertNotEqual(self.G1.listLivingMarried().len(), 3) marriedPeople = [] for indi in marriedPeople: self.assertIn(indi, self.G1.listLivingmarried()) #List all living people over 30 who have never been married in a GEDCOM file def test_US31_list_living_single(self): self.assertEqual(self.G1.listLivingSingle().len(), 5) self.assertNotEqual(self.G1.listLivingSingle().len(), 3) singlePeople = [] for indi in singlePeople: self.assertIn(indi, self.G1.listLivingSingle()) #List all multiple births in a GEDCOM file def test_US32_list_multiple_births(self): self.assertEqual(self.G1.listMultipleBirths().len(), 4) MultipleBirths = [] for birt in MultipleBirths: self.assertIn(birt, self.G1.listMultipleBirths()) #List all orphaned children (both parents dead and child < 18 years old) in a GEDCOM file def test_US33_list_orphans(self): self.assertEqual(self.G1.listOrphans().len(), 4) OrphansPeople = [] for indi in OrphansPeople: self.assertIn(indi, self.G1.listOrphans()) #List all couples who were married when the older spouse was more than twice as old as the younger spouse def test_US34_list_large_age_differences(self): self.assertEqual(self.G1.listLargeAgeDifferences().len(), 4) ageDifferences = [] for birt in ageDifferences: self.assertIn(birt, self.G1.listLargeAgeDifferences()) #List all people in a GEDCOM file who were born in the last 30 days def test_US35_list_recent_births(self): self.assertEqual(self.G1.listRecentBirths().len(), 5) self.assertNotEqual(self.G1.listRecentBirths().len(), 3) bornPeople = [] for indi in bornPeople: self.assertIn(indi, self.G1.listRecentBirths()) #list all people in a GEDCOM file who died in the last 30 days def test_US36_ListRecentDeaths(self): self.assertEqual(self.G1.listRecentDeaths().len(), 5) self.assertNotEqual(self.G1.listRecentDeaths().len(), 3) #manually input deceased people and append to the array deceasedProple = [] for indi in deceasedProple: self.assertIn(indi, self.G1.listRecentDeaths()) #list all living spouses and descendants of people in the GEDCOM who died in the last 30 days def test_US37_listRecentSurvivors(self): self.assertEqual(self.G1.listRecentSurviors().len(), 7) self.assertNotEqual(self.G1.listRecentSurviors().len(), 8) # manually input deceased people's relatives and append to the array deceasedProple = [] for indi in deceasedProple: self.assertIn(indi, self.G1.listRecentSurviors()) #list all living people in a GEDCOM file whose birthdays occur in the next 30 days def test_US38_listUpcomingBirthdays(self): self.assertEqual(self.G1.listUpcomingBirthdays().len(), 6) #manually input people with birthdays birthdayPeople = [] for indi in birthdayPeople: self.assertIn(indi, self.G1.listUpcomingBirthdays()) # list all living people in a GEDCOM file whose marriage anniversaries occur in the next 30 days def test_US39_UpcomingAnniversaries(self): self.assertEqual(self.G1.upcomingAnniversaries().len(), 4) #manually input individuals who have anniversaries coming up AnniversaryIndi = [] for indi in AnniversaryIndi: self.assertIn(indi, self.G1.upcomingAnniversaries()) # list line numbers from GEDCOM source file when reporting errors def test_US40_includeInputLineNumbers(self): self.assertEqual(self.G1.includeInputLineNumbers().len(), 2) self.assertTrue(self.G1.includeInputLineNumbers() == ['20', '25']) self.assertTrue(self.G2.includeInputLineNumbers() == ['15']) # Accept and use dates without days or without days and months def test_US41_IncludePartialDates(self): self.assertTrue(self.G1.IncludePartialDates()) # All dates should be legitimate dates for the months specified(e.g. 2/30/2015 is not legitimate) def test_US42_RejectIllegitimateDates(self): self.assertTrue(self.G1.rejectIllegitimateDates()) self.assertFalse(self.G2.rejectIllegitimateDates()) def testInputValidation(self): pass
def test_US28_Order_siblings_by_age(self): t1 = Family("t1") t2 = Family("t2") p1 = Individual("p1") p1.set_birthDate((1990, 4, 1)) p2 = Individual("p2") p2.set_birthDate((1990, 1, 1)) p3 = Individual("p3") p3.set_birthDate((1990, 9, 1)) p4 = Individual("p4") p4.set_birthDate((1987, 1, 1)) p5 = Individual("p5") p5.set_birthDate((2019, 1, 1)) p6 = Individual("p6") p6.set_birthDate((2017, 5, 30)) p7 = Individual("p7") p7.set_birthDate((2018, 3, 30)) p8 = Individual("p8") p8.set_birthDate((2019, 8, 30)) # -------------------------------------------------- t1.add_child(p1) t1.add_child(p2) t1.add_child(p3) t1.add_child(p4) t1.add_child(p5) t1.add_child(p6) t2.add_child(p1) t2.add_child(p2) t2.add_child(p3) t2.add_child(p4) t2.add_child(p7) t2.add_child(p8) # -------------------------------------------------- assert t1.order_siblings_by_age() == [p4, p2, p1, p3, p6, p5] assert t2.order_siblings_by_age() == [p4, p2, p1, p3, p7, p8]
def test_US12_parents_not_too_old(self): t1 = Family("t1") t2 = Family("t2") p1 = Individual("p1") p2 = Individual("p2") p3 = Individual("p3") p4 = Individual("p4") p5 = Individual("p5") p6 = Individual("p6") t1.set_wife(p1) t1.set_husband(p2) t1.add_child(p3) t2.set_wife(p4) t2.set_husband(p5) t2.add_child(p6) p1.set_birthDate(["1", "JAN", "1990"]) p2.set_birthDate(["1", "JAN", "1990"]) p4.set_birthDate(["1", "JAN", "1790"]) p5.set_birthDate(["1", "JAN", "1790"]) p3.set_birthDate(["1", "JAN", "2010"]) p6.set_birthDate(["1", "JAN", "2000"]) # --------------------------------- assert t1.parents_not_too_old() == True assert t2.parents_not_too_old() == False
def test_US14_multiple_births_lessOrEqual_than_5(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") p1 = Individual("p1") p1.set_birthDate(("1", "JAN", "1990")) p2 = Individual("p2") p2.set_birthDate(("1", "JAN", "1990")) p3 = Individual("p3") p3.set_birthDate(("1", "JAN", "1990")) p4 = Individual("p4") p4.set_birthDate(("3", "JAN", "1990")) p5 = Individual("p5") p5.set_birthDate(("2", "JAN", "1990")) p6 = Individual("p6") p6.set_birthDate(("30", "MAY", "1990")) p7 = Individual("p7") p7.set_birthDate(("2", "JAN", "1990")) p8 = Individual("p8") p8.set_birthDate(("2", "JAN", "1990")) p9 = Individual("p9") p9.set_birthDate(("2", "SEP", "1990")) p10 = Individual("p10") p10.set_birthDate(("2", "SEP", "1990")) p11 = Individual("p11") p11.set_birthDate(("3", "SEP", "1990")) p12 = Individual("p12") p12.set_birthDate(("3", "SEP", "1990")) p13 = Individual("p13") p13.set_birthDate(("3", "SEP", "1990")) # ------------------------------- t1.add_child(p1) t1.add_child(p2) t1.add_child(p3) t1.add_child(p4) t1.add_child(p5) t1.add_child(p6) t1.add_child(p7) t1.add_child(p8) t2.add_child(p1) t2.add_child(p2) t2.add_child(p3) t2.add_child(p4) t2.add_child(p5) t2.add_child(p6) t3.add_child(p3) t3.add_child(p4) t3.add_child(p5) t3.add_child(p6) t3.add_child(p7) t3.add_child(p8) t3.add_child(p9) t3.add_child(p10) t3.add_child(p11) t3.add_child(p12) t3.add_child(p13) # --------------------------------- assert t1.multiple_births_lessOrEqual_than_5() == False assert t2.multiple_births_lessOrEqual_than_5() == True assert t3.multiple_births_lessOrEqual_than_5() == False
def test_US13_siblings_spacing(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") t5 = Family("t5") t6 = Family("t6") p1 = Individual("p1") p1.set_birthDate(("1", "JAN", "1990")) p2 = Individual("p2") p2.set_birthDate(("1", "JAN", "1990")) p3 = Individual("p3") p3.set_birthDate(("1", "SEP", "1990")) p4 = Individual("p4") p4.set_birthDate(("2", "JAN", "1990")) p5 = Individual("p5") p5.set_birthDate(("3", "JAN", "1990")) p6 = Individual("p6") p6.set_birthDate(("30", "MAY", "1990")) # -------------------------------------------------- t1.add_child(p1) t1.add_child(p2) t2.add_child(p1) t2.add_child(p3) t3.add_child(p1) t3.add_child(p4) t4.add_child(p1) t4.add_child(p5) t5.add_child(p1) t5.add_child(p6) t6.add_child(p1) t6.add_child(p3) t6.add_child(p6) # -------------------------------------------------- assert t1.siblings_spacing() == True assert t2.siblings_spacing() == True assert t3.siblings_spacing() == True assert t4.siblings_spacing() == False assert t5.siblings_spacing() == False assert t6.siblings_spacing() == False
def test_US10_marriage_after_14(self): t1 = Family("F01") male1 = Individual("P01") female1 = Individual("P02") male1.set_birthDate(['8', 'SEP', '2000']) female1.set_birthDate(['8', 'SEP', '2000']) t1.set_husband(male1) t1.set_wife(female1) t1.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t2 = Family("F02") male2 = Individual("P03") female2 = Individual("P04") male2.set_birthDate(['7', 'SEP', '2000']) female2.set_birthDate(['8', 'SEP', '2000']) t2.set_husband(male2) t2.set_wife(female2) t2.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t3 = Family("F03") male3 = Individual("P05") female3 = Individual("P06") male3.set_birthDate(['8', 'SEP', '2000']) female3.set_birthDate(['7', 'SEP', '2000']) t3.set_husband(male3) t3.set_wife(female3) t3.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t4 = Family("F04") male4 = Individual("P07") female4 = Individual("P08") male4.set_birthDate(['1', 'SEP', '1990']) female4.set_birthDate(['2', 'SEP', '1990']) t4.set_husband(male4) t4.set_wife(female4) t4.set_marriedDate(['8', 'SEP', '2014']) # -------------------------------------------------- t5 = Family("F05") male5 = Individual("P09") female5 = Individual("P10") male5.set_birthDate(['09', 'APR', '1997']) female5.set_birthDate(['19', 'DEC', '1997']) t5.set_husband(male5) t5.set_wife(female5) t5.set_marriedDate(['1', 'JUN', '2007']) # -------------------------------------------------- assert t1.marriage_after_14() == False assert t2.marriage_after_14() == False assert t3.marriage_after_14() == False assert t4.marriage_after_14() == True assert t5.marriage_after_14() == False
def parse(self): #from Individual import Individual #from Family import Family offset = 0 for i in range(len(self._data[1]) - 1): # enumerate individuals start_index = self._data[1][i] end_index = self._data[1][i + 1] # print(self._data[0][start_index:end_index]) # TODO: check deplicated ids id = self._data[0][start_index][2] if id in self._individuals: offset += 1 id = "@I" + str(int(id[2:-1]) + offset) + "@" new_indi = Individual(id) new_indi.set_lineNum(self._data[3][start_index]) self._individuals[id] = new_indi # print(start_index, end_index) for j in range(start_index + 1, end_index): level, tag, arguments = self._data[0][j] if tag == "NAME": new_indi.set_name("".join(arguments)) elif tag == "SEX": new_indi.set_gender(arguments[0]) elif tag == "BIRT": # set missing dates to Jan/01 j += 1 level, tag, arguments = self._data[0][j] try: new_indi.set_birthDate(arguments) except ValueError as e: print(e) # TODO:how to handle error? elif tag == "DEAT": j += 1 level, tag, arguments = self._data[0][j] new_indi.set_deathDate(arguments) elif tag == "FAMS": if arguments[0] not in self._families: new_fam = Family(arguments[0]) self._families[arguments[0]] = new_fam new_indi.add_to_family(self._families[arguments[0]]) elif tag == "FAMC": if arguments[0] not in self._families: new_fam = Family(arguments[0]) self._families[arguments[0]] = new_fam new_indi.set_parentFamily(self._families[arguments[0]]) offset = 0 for i in range(len(self._data[2]) - 1): start_index = self._data[2][i] end_index = self._data[2][i + 1] # print(self._data[0][start_index:end_index]) id = self._data[0][start_index][2] # if id in self._families: #TODO: handle duplicate # offset+=1 # id = "@F" + str(int(id[2:-1]) + offset) + "@" new_fam = Family(id) new_fam.set_lineNum(self._data[3][start_index]) self._families[id] = new_fam for j in range(start_index + 1, end_index): level, tag, arguments = self._data[0][j] # print(level, tag, arguments) if tag == "HUSB": new_fam.set_husband(self._individuals[arguments[0]]) elif tag == "WIFE": new_fam.set_wife(self._individuals[arguments[0]]) elif tag == "CHIL": new_fam.add_child(self._individuals[arguments[0]]) elif tag == "MARR": j += 1 level, tag, arguments = self._data[0][j] # print(level, tag, arguments) try: new_fam.set_marriedDate(arguments) except: continue elif tag == "DIV": j += 1 level, tag, arguments = self._data[0][j] # print(level, tag, arguments) try: new_fam.set_divorcedDate(arguments) except: continue
def test_siblings_spacing(self): t1 = Family("t1") t2 = Family("t2") t3 = Family("t3") t4 = Family("t4") t5 = Family("t5") t6 = Family("t6") p1 = Individual("p1") p1.set_birthDate((1990, 1, 1)) p2 = Individual("p2") p2.set_birthDate((1990, 1, 1)) p3 = Individual("p3") p3.set_birthDate((1990, 9, 1)) p4 = Individual("p4") p4.set_birthDate((1990, 1, 2)) p5 = Individual("p5") p5.set_birthDate((1990, 1, 3)) p6 = Individual("p6") p6.set_birthDate((1990, 5, 30)) # -------------------------------------------------- t1.add_child(p1) t1.add_child(p2) t2.add_child(p1) t2.add_child(p3) t3.add_child(p1) t3.add_child(p4) t4.add_child(p1) t4.add_child(p5) t5.add_child(p1) t5.add_child(p6) t6.add_child(p1) t6.add_child(p3) t6.add_child(p6) # -------------------------------------------------- assert t1.siblings_spacing() == True assert t2.siblings_spacing() == True assert t3.siblings_spacing() == True assert t4.siblings_spacing() == False assert t5.siblings_spacing() == False assert t6.siblings_spacing() == False