]
_families = [
Family('F1',
husband_id='1',
wife_id='2',
children='3',
married=datetime(day=1, month=1, year=1948)),
Family('F2',
husband_id='5',
wife_id='6',
children='4',
married=datetime(day=1, month=1, year=1963)),
]
parents_age_okay = Tree()
[parents_age_okay.add_individual(i) for i in _individuals]
[parents_age_okay.add_family(f) for f in _families]
_individuals2 = [
Individual('1',
name='Michelle /Obama/',
sex='M',
birthday=datetime(day=20, month=1, year=1947)),
Individual('2',
name='Barack /Obama/',
sex='F',
birthday=datetime(day=20, month=2, year=1948)),
Individual('3',
name='Bill /Clinton/',
name='Jane /Doe/',
sex='M',
birthday=datetime(day=2, month=3, year=1962)),
Individual('e',
name='Ava /White/',
sex='M',
birthday=datetime(day=1, month=4, year=1985)),
Individual('f',
name='Maria /White/',
sex='F',
birthday=datetime(day=9, month=10, year=1988)),
Individual('g',
name='Ken /White/',
sex='F',
birthday=datetime(day=30, month=1, year=1946)),
Individual('h',
name='Ana /White/',
sex='M',
birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='c', wife_id='d', children=['1', '2']),
Family('F2', husband_id='g', wife_id='h', children=['5', '6'])
]
two_husb_and_wife_incorrect = Tree()
[two_husb_and_wife_incorrect.add_individual(i) for i in _individuals]
[two_husb_and_wife_incorrect.add_family(f) for f in _families]
from datetime import datetime
from lib.GedObjects import Tree
from lib.GedObjects import Family
from lib.GedObjects import Individual
_individuals = [
Individual('1a1a', name='Evan /Doe/', sex='M', birthday=datetime(day=12, month=4, year=1998)),
Individual('2b2b', name='Elena /Doe/', sex='F', birthday=datetime(day=15, month=9, year=2000)),
Individual('3c3c', name='John /Doe/', sex='F', birthday=datetime(day=2, month=3, year=1962)),
Individual('4d4d', name='Jane /Doe/', sex='F', birthday=datetime(day=2, month=3, year=1962)),
Individual('5e5e', name='Ava /White/', sex='M', birthday=datetime(day=1, month=4, year=1985)),
Individual('6f6f', name='Maria /White/', sex='F', birthday=datetime(day=9, month=10, year=1988)),
Individual('7g7g', name='Ken /White/', sex='M', birthday=datetime(day=30, month=1, year=1946)),
Individual('8h8h', name='Ana /White/', sex='F', birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3c3c', wife_id='4d4d', children=['1', '2']),
Family('F2', husband_id='7g7g', wife_id='8h8h', children=['5', '6'])
]
husband_incorrect = Tree()
[husband_incorrect .add_individual(i) for i in _individuals]
[husband_incorrect .add_family(f) for f in _families]
from datetime import datetime, timedelta
import lib.GedConstants as gc
birth1 = datetime.today()
birth2 = datetime.today() - timedelta(days=1)
birth3 = datetime.today() - timedelta(days=29)
birth4 = datetime.today() - timedelta(days=15)
_recently_born_individuals = [
Individual('1', name='Randy /Pay/', birthday=birth1),
Individual('2', name='Rondy /Pay/', birthday=birth2),
Individual('3', name='Rendy /Day/', birthday=birth3),
Individual('4', name='Rindy /Day/', birthday=birth4)
]
all_recent_births_tree = Tree()
[all_recent_births_tree.add_individual(i) for i in _recently_born_individuals]
birth5 = datetime.today()
birth6 = datetime.today() - timedelta(days=40)
birth7 = datetime.today() - timedelta(days=27)
birth8 = datetime.today() - timedelta(days=1000)
some_recently_born_individuals = [
Individual('5', name='Mickey /Mouse/', birthday=birth5),
Individual('6', name='Minnie /Mouse/', birthday=birth6),
Individual('7', name='John /Mouse/', birthday=birth7),
Individual('8', name='Jane /Mouse/', birthday=birth8)
]
sex='M',
birthday=datetime(day=8, month=3, year=1970),
death=datetime(day=23, month=5, year=2018)),
Individual('4',
name='Leslie /Maple/',
sex='F',
birthday=datetime(day=12, month=9, year=1950))
]
_families = [
Family('F1', married=datetime(1973, 4, 11, 0, 0)),
Family('F2',
married=datetime(2000, 10, 31, 0, 0),
divorced=datetime(2007, 12, 25, 0, 0))
]
correct_dates = Tree()
[correct_dates.add_individual(i) for i in _individuals]
[correct_dates.add_family(f) for f in _families]
"""Tree for Test Case 2"""
_individuals1 = [
Individual('1',
name='Super /Man/',
sex='M',
birthday=datetime(day=28, month=1, year=1954),
death=datetime(day=5, month=11, year=2017)),
Individual('2',
name='Angela /Brown/',
sex='F',
birthday=datetime(day=16, month=2, year=1987)),
Individual('3',
name='Jimmy /Gold/',
sex='F',
birthday=datetime(day=2, month=3, year=1962)),
Individual('5',
name='Ava /White/',
sex='F',
birthday=datetime(day=1, month=4, year=1985),
child='F2'),
Individual('6',
name='Maria /White/',
sex='F',
birthday=datetime(day=9, month=10, year=1988),
child='F2'),
Individual('7',
name='Ken /White/',
sex='M',
birthday=datetime(day=30, month=1, year=1946)),
Individual('8',
name='Ana /White/',
sex='F',
birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3', wife_id='4', children=['1', '2']),
Family('F2', husband_id='7', wife_id='8', children=['5', '6'])
]
father_diff = Tree()
[father_diff.add_individual(i) for i in _individuals]
[father_diff.add_family(f) for f in _families]
from datetime import datetime
from lib.GedObjects import Tree
from lib.GedObjects import Family
from lib.GedObjects import Individual
_individuals = [
Individual('1', name='Evan /Doe/', sex='M', birthday=datetime(day=12, month=4, year=1998)),
Individual('2', name='Elena /Doe/', sex='F', birthday=datetime(day=15, month=9, year=2000)),
Individual('3', name='John /Doe/', sex='M', birthday=datetime(day=2, month=3, year=1962)),
Individual('4', name='Jane /Doe/', sex='F', birthday=datetime(day=2, month=3, year=1962)),
Individual('5', name='Ava /White/', sex='M', birthday=datetime(day=1, month=4, year=1985)),
Individual('6', name='Maria /White/', sex='F', birthday=datetime(day=9, month=10, year=1988)),
Individual('7', name='Ken /White/', sex='M', birthday=datetime(day=30, month=1, year=1946)),
Individual('8', name='Ana /White/', sex='F', birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3', wife_id='4', children=['1', '2'])
]
all_correct_gender = Tree()
[all_correct_gender .add_individual(i) for i in _individuals]
[all_correct_gender .add_family(f) for f in _families]
birthday=datetime(1968, 2, 10, 0, 0),
death=datetime(1969, 3, 10, 0, 0),
spouse=None),
Individual('7',
name='Tindy /May/',
birthday=datetime(1969, 3, 10, 0, 0),
death=datetime(1970, 4, 10, 0, 0),
spouse=None),
Individual('8',
name='Tendy /May/',
birthday=datetime(2005, 4, 10, 0, 0),
death=None,
spouse=None)
]
all_correct_tree = Tree()
some_spouse_tree = Tree()
some_under_tree = Tree()
some_dead_tree = Tree()
all_fail_tree = Tree()
[all_correct_tree.add_individual(i) for i in _individuals1]
[some_spouse_tree.add_individual(i) for i in _individuals2]
[some_under_tree.add_individual(i) for i in _individuals3]
[some_dead_tree.add_individual(i) for i in _individuals4]
[all_fail_tree.add_individual(i) for i in _individuals5]
class TestLivingandSingle(unittest.TestCase):
def test_all_correct(self):
self.assertEqual(
# some are dead
Individual('1', name='Randy /Pay/', birthday=today + timedelta(days=45), death=datetime(1964, 6, 10, 0, 0)),
Individual('2', name='Rondy /Pay/', birthday=today + timedelta(days=2), death=None),
Individual('3', name='Rendy /Day/', birthday=today + timedelta(days=4), death=None),
Individual('4', name='Rindy /Day/', birthday=today + timedelta(days=40), death=datetime(1967, 1, 10, 0, 0))
]
_individuals5 = [
# all should fail
Individual('1', name='Randy /Pay/', birthday=today + timedelta(days=40), death=datetime(1964, 6, 10, 0, 0)),
Individual('2', name='Rondy /Pay/', birthday=today + timedelta(days=42), death=None),
Individual('3', name='Rendy /Day/', birthday=today + timedelta(days=43), death=None),
Individual('4', name='Rindy /Day/', birthday=today + timedelta(days=44), death=datetime(1967, 1, 10, 0, 0))
]
all_upcoming_tree = Tree()
some_upcoming_tree = Tree()
no_birth_tree = Tree()
some_dead_tree = Tree()
no_upcoming_tree = Tree()
[all_upcoming_tree.add_individual(i) for i in _individuals1]
[some_upcoming_tree.add_individual(i) for i in _individuals2]
[no_birth_tree.add_individual(i) for i in _individuals3]
[some_dead_tree.add_individual(i) for i in _individuals4]
[no_upcoming_tree.add_individual(i) for i in _individuals5]
class TestUpBirth(unittest.TestCase):
def test_all_correct(self):
self.assertEqual([res[:-1] for res in all_upcoming_tree.upcoming_birthday()],
sex='F',
birthday=datetime(day=2, month=3, year=1962)),
Individual('5',
name='Ava /White/',
sex='F',
birthday=datetime(day=1, month=4, year=1985),
child='F2'),
Individual('6',
name='Maria /White/',
sex='F',
birthday=datetime(day=9, month=10, year=1988),
child='F2'),
Individual('7',
name='Ken /White/',
sex='M',
birthday=datetime(day=30, month=1, year=1946)),
Individual('8',
name='Ana /White/',
sex='F',
birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3', wife_id='4', children=['1', '2']),
Family('F2', husband_id='7', wife_id='8', children=['5', '6'])
]
son_diff = Tree()
[son_diff.add_individual(i) for i in _individuals]
[son_diff.add_family(f) for f in _families]
name='Jane /Doe/',
sex='M',
birthday=datetime(day=2, month=3, year=1962)), # 4444 is wrong
Individual('5555',
name='Ava /White/',
sex='M',
birthday=datetime(day=1, month=4, year=1985)),
Individual('6666',
name='Maria /White/',
sex='F',
birthday=datetime(day=9, month=10, year=1988)),
Individual('7777',
name='Ken /White/',
sex='M',
birthday=datetime(day=30, month=1, year=1946)),
Individual('8888',
name='Ana /White/',
sex='F',
birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3333', wife_id='4444', children=['1', '2']),
Family('F2', husband_id='7777', wife_id='8888', children=['5', '6'])
]
wife_incorrect = Tree()
[wife_incorrect.add_individual(i) for i in _individuals]
[wife_incorrect.add_family(f) for f in _families]
birthday=datetime(1965, 7, 10, 0, 0),
death=None,
spouse=None),
Individual('4',
name='Rindy /Day/',
birthday=datetime(1966, 8, 10, 0, 0),
death=datetime(1967, 1, 10, 0, 0),
spouse=None),
Individual('5',
name='Rundy /Lay/',
birthday=datetime(1967, 1, 10, 0, 0),
death=None,
spouse=None),
]
all_alive_tree = Tree()
some_dead_tree = Tree()
[all_alive_tree.add_individual(i) for i in _individuals1]
[some_dead_tree.add_individual(i) for i in _individuals2]
class TestLivingandSinglListDead(unittest.TestCase):
def test_all_alive(self):
self.assertEqual([res[0] for res in all_alive_tree.list_deceased()],
[])
def test_some_dead(self):
self.assertEqual([res[0] for res in some_dead_tree.list_deceased()],
['Randy /Pay/', 'Rindy /Day/'])
def evaluate(self, tree: Tree, level: int, tag: str, args: str):
"""
Evaluate a valid line and add information to the tree if necessary
:param tree: The Tree currently being built
:param level: (unused) The level of the line to evaluate
:param tag: The tag of the line to evaluate
:param args: The args of the line to evaluate
:return: None
"""
# If we are creating an individual and the next tag is invalid for an individual
if self._creating_indi and (tag in Builder.FAMILY_TAGS
or tag in Builder.TOP_LEVEL_TAGS):
self._creating_indi = False
# Try to add the individual
try:
tree.add_individual(Individual(**self._current_indi_data))
except (TypeError, DuplicateIndividualException):
logging.log(logging.DEBUG, traceback.format_exc())
self._current_indi_data = {}
# If we are creating a family and the next tag is invalid for a family
elif self._creating_fam and (tag in Builder.INDIVIDUAL_TAGS
or tag in Builder.TOP_LEVEL_TAGS):
self._creating_fam = False
# Try to add the family
try:
tree.add_family(Family(**self._current_fam_data))
except (TypeError, DuplicateFamilyException):
logging.log(logging.DEBUG, traceback.format_exc())
self._current_fam_data = {'children': []}
# If we are creating some kind of date and the next tag is not 'DATE' or is invalid
elif True in (self._creating_birth, self._creating_death,
self._creating_marr,
self._creating_div) and (tag != 'DATE'):
self._creating_birth, self._creating_death, self._creating_marr, self._creating_div = False, False, False, False
# Basic cases for each tag
if tag == 'INDI':
self._creating_indi = True
self._current_indi_data['id'] = args
elif tag == 'FAM':
self._creating_fam = True
self._current_fam_data['id'] = args
elif tag == 'NAME' and self._creating_indi:
self._current_indi_data['name'] = args
elif tag == 'SEX' and self._creating_indi:
self._current_indi_data['sex'] = args
elif tag == 'BIRT' and self._creating_indi:
self._creating_birth = True
elif tag == 'DEAT' and self._creating_indi:
self._creating_death = True
elif tag == 'MARR' and self._creating_fam:
self._creating_marr = True
elif tag == 'DIV' and self._creating_fam:
self._creating_div = True
elif tag == 'DATE':
timestamp = datetime.strptime(args, '%d %b %Y')
if self._creating_birth:
self._current_indi_data['birthday'] = timestamp
elif self._creating_death:
self._current_indi_data['death'] = timestamp
elif self._creating_marr:
self._current_fam_data['married'] = timestamp
elif self._creating_div:
self._current_fam_data['divorced'] = timestamp
elif tag == 'FAMC' and self._creating_indi:
self._current_indi_data['child'] = args
elif tag == 'FAMS' and self._creating_indi:
self._current_indi_data['spouse'] = args
elif tag == 'HUSB' and self._creating_fam:
self._current_fam_data['husband_id'] = args
elif tag == 'WIFE' and self._creating_fam:
self._current_fam_data['wife_id'] = args
elif tag == 'CHIL' and self._creating_fam:
self._current_fam_data['children'].append(args)
elif tag == 'HEAD':
pass
elif tag == 'TRLR':
pass
elif tag == 'NOTE':
pass
def build_tree(self, filename: str):
tree = Tree()
for level, tag, args, valid in _parse_file(filename):
if valid:
self.evaluate(tree, level, tag, args)
return tree
from datetime import datetime
from lib.GedObjects import Tree
from lib.GedObjects import Family
from lib.GedObjects import Individual
_individuals = [
Individual('1', name='Evan /Doe/', sex='M', birthday=datetime(day=12, month=4, year=1998), child='F1'),
Individual('2', name='Elena /Doe/', sex='F', birthday=datetime(day=15, month=9, year=2000), child='F1'),
Individual('9', name='Maxwell /Baker/', sex='M', birthday=datetime(day=2, month=3, year=1997), child='F1'),
Individual('10', name='Kevin /Tan/', sex='M', birthday=datetime(day=12, month=8, year=2002), child='F1'),
Individual('3', name='John /Doe/', sex='M', birthday=datetime(day=2, month=3, year=1962)),
Individual('4', name='Jane /Doe/', sex='F', birthday=datetime(day=2, month=3, year=1962)),
Individual('5', name='Ava /White/', sex='F', birthday=datetime(day=1, month=4, year=1985), child='F2'),
Individual('6', name='Joseph /Tanner/', sex='M', birthday=datetime(day=9, month=10, year=1988), child='F2'),
Individual('7', name='Ken /White/', sex='M', birthday=datetime(day=30, month=1, year=1946)),
Individual('8', name='Ana /White/', sex='F', birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3', wife_id='4', children=['1', '2', '9', '10']),
Family('F2', husband_id='7', wife_id='8', children=['5', '6'])
]
two_fams_abb = Tree()
[two_fams_abb .add_individual(i) for i in _individuals]
[two_fams_abb .add_family(f) for f in _families]
sex='F',
birthday=datetime(day=2, month=3, year=1962)),
Individual('5',
name='Ava /White/',
sex='F',
birthday=datetime(day=1, month=4, year=1985),
child='F2'),
Individual('6',
name='Maria /White/',
sex='F',
birthday=datetime(day=9, month=10, year=1988),
child='F2'),
Individual('7',
name='Ken /White/',
sex='M',
birthday=datetime(day=30, month=1, year=1946)),
Individual('8',
name='Ana /White/',
sex='F',
birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3', wife_id='4', children=['1', '2']),
Family('F2', husband_id='7', wife_id='8', children=['5', '6'])
]
all_same_names = Tree()
[all_same_names.add_individual(i) for i in _individuals]
[all_same_names.add_family(f) for f in _families]
birthday=datetime(1972, 10, 17, 0, 0),
death=datetime(1999, 8, 26, 0, 0),
spouse=None),
Individual('5',
name='Jane /Name/',
birthday=datetime(1987, 11, 8, 0, 0),
death=None,
spouse=None),
Individual('6',
name='John /Name/',
birthday=datetime(1986, 12, 31, 0, 0),
death=datetime(2007, 4, 19, 0, 0),
spouse='F3')
]
all_correct_tree = Tree()
some_correct_tree = Tree()
all_fail_tree = Tree()
[all_correct_tree.add_individual(i) for i in _individuals1]
[all_correct_tree.add_family(f) for f in _families]
[some_correct_tree.add_individual(i) for i in _individuals2]
[some_correct_tree.add_family(f) for f in _families]
[all_fail_tree.add_individual(i) for i in _individuals4]
[all_fail_tree.add_family(f) for f in _families]
class TestLivingandSingle(unittest.TestCase):
def test_all_correct(self):
self.assertEqual(
[res[:-1] for res in all_correct_tree.list_large_age_difference()],
husband_id='1',
wife_id='2',
children='2',
married=datetime(day=1, month=1, year=1948)),
Family('F2',
husband_id='3',
wife_id='4',
married=datetime(day=1, month=1, year=1963)),
Family('F3',
husband_id='5',
wife_id='6',
children='5',
married=datetime(day=1, month=1, year=1951)),
]
in_wed = Tree()
[in_wed.add_individual(i) for i in _individuals]
[in_wed.add_family(f) for f in _families]
_individuals2 = [
Individual('a',
name='Toddy /Frod/',
sex='M',
birthday=datetime(day=20, month=1, year=1947)),
Individual('b',
name='Loddy /Frod/',
sex='F',
birthday=datetime(day=20, month=2, year=1948)),
Individual('c',
name='Moosy /Goose/',
_families = [
Family('F1',
husband_id='1',
wife_id='2',
married=datetime(day=1, month=1, year=1948)),
Family('F2',
husband_id='3',
wife_id='4',
married=datetime(day=1, month=1, year=1963)),
Family('F3',
husband_id='5',
wife_id='6',
married=datetime(day=1, month=1, year=1951)),
]
born_wed = Tree()
[born_wed.add_individual(i) for i in _individuals]
[born_wed.add_family(f) for f in _families]
_individuals2 = [
Individual('a',
name='Toddy /Frod/',
sex='M',
birthday=datetime(day=20, month=1, year=1947)),
Individual('b',
name='Loddy /Frod/',
sex='F',
birthday=datetime(day=20, month=2, year=1948)),
Individual('c',
name='Moosy /Goose/',
name='Fonty /Python/',
sex='F',
birthday=datetime(day=23, month=3, year=1999)),
Individual('5',
name='Lonty /Python/',
sex='M',
birthday=datetime(day=23, month=3, year=1999)),
Individual('6',
name='Zonty /Python/',
sex='F',
birthday=datetime(day=23, month=3, year=1999)),
]
_families = [Family('F1', children=['1', '2', '3', '4', '5', '6'])]
sextuplet_check = Tree()
[sextuplet_check.add_individual(i) for i in _individuals]
[sextuplet_check.add_family(f) for f in _families]
_individuals2 = [
Individual('1',
name='Monty /Python//',
sex='M',
birthday=datetime(day=23, month=3, year=1999)),
Individual('2',
name='Ronty /Python/',
sex='F',
birthday=datetime(day=23, month=3, year=1998)),
Individual('3',
name='Tonty /Python/',
sex='F',
birthday=datetime(day=2, month=3, year=1962)),
Individual('5',
name='Ava /White/',
sex='F',
birthday=datetime(day=1, month=4, year=1985),
child='F2'),
Individual('6',
name='Maria /White/',
sex='F',
birthday=datetime(day=9, month=10, year=1988),
child='F2'),
Individual('7',
name='Ken /White/',
sex='M',
birthday=datetime(day=30, month=1, year=1946)),
Individual('8',
name='Ana /White/',
sex='F',
birthday=datetime(day=3, month=5, year=1940))
]
_families = [
Family('F1', husband_id='3', wife_id='4', children=['1', '2', '9', '10']),
Family('F2', husband_id='7', wife_id='8', children=['5', '6'])
]
three_sons_one_diff = Tree()
[three_sons_one_diff.add_individual(i) for i in _individuals]
[three_sons_one_diff.add_family(f) for f in _families]
Individual('9',
name='Snow /White',
sex='F',
birthday=datetime(day=6, month=1, year=1996)),
Individual('10',
name='Prince /Charming',
sex='M',
birthday=datetime(day=13, month=4, year=1993))
]
_families = [
Family('F1',
husband_id='1',
wife_id='2',
children=['3'],
married=datetime(day=1, month=1, year=1963)),
Family('F2',
husband_id='4',
wife_id='5',
children=['6', '7', '8'],
married=datetime(day=3, month=5, year=1967)),
Family('F3',
husband_id='10',
wife_id='9',
married=datetime(day=24, month=12, year=2018))
]
of_age = Tree()
[of_age.add_individual(i) for i in _individuals]
[of_age.add_family(f) for f in _families]
