def two_pair_twins_two_families(self): individuals = { 'CHILD1': Individual(id='CHILD1', birth=date(2005, 3, 10), famc='FAM1'), 'CHILD2': Individual(id='CHILD2', birth=date(2006, 2, 23), famc='FAM1'), 'CHILD3': Individual(id='CHILD3', birth=date(2009, 7, 20), famc='FAM1'), 'CHILD4': Individual(id='CHILD4', birth=date(2009, 7, 20), famc='FAM1'), 'CHILD5': Individual(id='CHILD5', birth=date(2007, 5, 5), famc='FAM2'), 'CHILD6': Individual(id='CHILD6', birth=date(2007, 5, 5), famc='FAM2') } families = { 'FAM1': Family(id='FAM1', child_ids=['CHILD1', 'CHILD2', 'CHILD3', 'CHILD4']), 'FAM2': Family(id='FAM2', child_ids=['CHILD5', 'CHILD6']) } parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US32.list_multiple_births(parser, db) self.assertEqual(len(pt_list), 1) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 2)
def living_married(self): individuals = { "HUSB1": Individual(id="HUSB1", fams="FAM1"), "WIFE1": Individual(id="WIFE1", fams="FAM1"), "HUSB2": Individual(id="HUSB2", fams="FAM2"), "WIFE2": Individual(id="WIFE2", fams="FAM2"), "WIFE3": Individual(id="WIFE3", famc="FAM3"), } families = { "FAM1": Family(id="FAM1", married=date(2012, 11, 12), divorced=None, husband_id="HUSB1", wife_id="WIFE1"), "FAM2": Family(id="FAM2", married=date(2012, 10, 16), divorced=None, husband_id="HUSB2", wife_id="WIFE2"), } parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US30.list_living_married(parser, db) #print(pt_list) self.assertGreater(TableHelpers.get_row_count(pt_list[0]), 0)
def nullDeath(self): individuals = { "NULLBIRT": Individual(id="NULLBIRT", death=None) } parser = TestParser(individuals, {}) db = TestDatabase(individuals, {}) pt_list = US36.list_recent_deaths(parser, db) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 0)
def recent(self): individuals = { "RECENT": Individual(id="RECENT", death=date.today()), "OLD": Individual(id="OLD", death=date(1989,7,31)) } parser = TestParser(individuals, {}) db = TestDatabase(individuals, {}) pt_list = US36.list_recent_deaths(parser, db) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 1)
def noneRecent(self): individuals = { "OLD1": Individual(id="OLD1", death=date(1990,2,22)), "OLD2": Individual(id="OLD2", death=date(1991,6,16)) } parser = TestParser(individuals, {}) db = TestDatabase(individuals, {}) pt_list = US36.list_recent_deaths(parser, db) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 0)
def two_anniversaries(self): marriage = date(2001, self.today.month, self.today.day) individuals = {} families = { "FAM1": Family(id="FAM1", married=DateHelpers.add_time(marriage, 15, "days")), "FAM2": Family(id="FAM2", married=DateHelpers.add_time(marriage, 16, "days")), "FAM3": Family(id="FAM3", married=DateHelpers.add_time(marriage, 100, "days")), } parser = TestParser(individuals, families) pt_list = US39.upcoming_anniversaries(parser) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 2)
def all_deceased(self): individuals = { "PERSON1": Individual(id="PERSON1", death=date(2005, 10, 5)), "PERSON2": Individual(id="PERSON2", death=date(2005, 10, 15)), "PERSON3": Individual(id="PERSON3", death=date(2006, 9, 5)), } families = {} parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US29.list_deceased(parser, db) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 3)
def none_deceased(self): individuals = { "PERSON1": Individual(id="PERSON1"), "PERSON2": Individual(id="PERSON2"), "PERSON3": Individual(id="PERSON3"), } families = {} parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US29.list_deceased(parser, db) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 0)
def large_age_differences(self): individuals = { "HUSB1": Individual(id="HUSB1", birth=date(1967,9,30),fams="FAM1"), "WIFE1": Individual(id="WIFE1", birth=date(1989,7,31),fams="FAM1") } families = { "FAM1": Family(id="FAM1", married=date(2009,10,31), divorced=None, husband_id="HUSB1", wife_id="WIFE1") } parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US34.list_large_age_differences(parser, db) #print(pt_list[0]) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 2)
def not_large_age_differences(self): individuals = { "HUSB2": Individual(id="HUSB2", birth=date(1990,2,22),fams="FAM2"), "WIFE2": Individual(id="WIFE2", birth=date(1991,6,16),fams="FAM2") } families = { "FAM2": Family(id="FAM2", married=date(2012,10,16), divorced=None, husband_id="HUSB2", wife_id="WIFE2") } parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US34.list_large_age_differences(parser, db) #print(pt_list[0]) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 0)
def living_single(self): individuals = { "HUSB": Individual(id="HUSB", birth=date(1900, 2, 11), famc="FAM1"), "WIFE": Individual(id="WIFE", birth=date(1930, 4, 18), famc="FAM2"), "HUSB2": Individual(id="HUSB2", birth=date(2000, 8, 17), fams="FAM2"), "WIFE2": Individual(id="WIFE2", birth=date(1990, 9, 26), fams="FAM2"), } families = { "FAM2": Family(id="FAM2", married=date(2012,10,16), divorced=None, husband_id="HUSB2", wife_id="WIFE2") } parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US31.list_living_single(parser, db) #print(pt_list) self.assertGreater(TableHelpers.get_row_count(pt_list[0]), 0)
def two_birthdays(self): birthday = date(2001, self.today.month, self.today.day) individuals = { "PERSON1": Individual(id="PERSON1", birth=DateHelpers.add_time(birthday, 25, "days")), "PERSON2": Individual(id="PERSON2", birth=DateHelpers.add_time(birthday, 70, "days")), "PERSON3": Individual(id="PERSON3", birth=DateHelpers.add_time(birthday, 15, "days")), } families = {} parser = TestParser(individuals, families) pt_list = US38.upcoming_birthdays(parser) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 2)
def one_pair_twins(self): individuals = { 'CHILD1': Individual(id='CHILD1', birth=date(2006, 1, 15), famc='FAM1'), 'CHILD2': Individual(id='CHILD2', birth=date(2007, 6, 19), famc='FAM1'), 'CHILD3': Individual(id='CHILD3', birth=date(2007, 6, 19), famc='FAM1') } families = { 'FAM1': Family(id='FAM1', child_ids=['CHILD1', 'CHILD2', 'CHILD3']) } parser = TestParser(individuals, families) db = TestDatabase(individuals, families) pt_list = US32.list_multiple_births(parser, db) self.assertEqual(len(pt_list), 1) self.assertEqual(TableHelpers.get_row_count(pt_list[0]), 1)