def test_erroneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us07_less_than_150_years_old.ged")
indivs_150 = validate.less_than_150_years_old(indivs_df)
expected = {
'ID': {
0: '@ani@',
2: '@luke@'
},
'NAME': {
0: 'Anakin /Skywalker/',
2: 'Luke /Skywalker/'
},
'BIRTHDAY': {
0: '25 MAY 1777',
2: '26 SEP 1865'
},
'DEATH': {
0: '19 MAY 2005',
2: np.nan
},
'AGE': {
0: 227.0,
2: 153.0
}
}
self.assertEqual(
indivs_150[['ID', 'NAME', 'BIRTHDAY', 'DEATH', 'AGE']].to_dict(),
expected)
def test_get_spouses(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/utils_test_get_descendents.ged")
self.assertEqual({'@mystery@', '@cliegg@'},
utils.get_spouses("@shmi@", fams_df))
self.assertEqual({'@lea@'}, utils.get_spouses("@han@", fams_df))
self.assertEqual(set(), utils.get_spouses("@luke@", fams_df))
def test_erroneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us15_fewer_than_15_siblings.ged")
fams_error = validate.fewer_than_15_siblings(indivs_df, fams_df)
expected = ['@sky1@']
actual = fams_error["ID"].values.tolist()
self.assertEqual(expected, actual)
def test_empty_gedcom(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/parser_test_empty.ged")
self.assertEqual(list(indivs_df.columns), indivs_columns)
self.assertEqual(list(fams_df.columns), fams_columns)
self.assertTrue(indivs_df.empty)
self.assertTrue(fams_df.empty)
def test_erroneous(self):
indiv_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us16_male_last_name.ged")
fams_error = validate.same_male_last_name(indiv_df, fams_df)
expected = ['@ani1@', '@ani2@']
actual = fams_error["ID"].values.tolist()
self.assertEqual(expected, actual)
def test_dead_people(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/us29_list_dead_people.ged")
violations = validate.list_deceased(indivs_df)
expected = {
'ID': {
1: '@shmi@',
2: '@cliegg@',
3: '@owen@',
4: '@ani@'
},
'BIRTHDAY': {
1: '25 MAY 1957',
2: '25 MAY 1952',
3: '25 MAY 1978',
4: '25 MAY 1977'
},
'DEATH': {
1: '16 MAY 2002',
2: '28 OCT 2018',
3: '19 MAY 2005',
4: '19 MAY 2005'
}
}
self.assertEqual(violations[['ID', 'BIRTHDAY', 'DEATH']].to_dict(),
expected)
def test(self):
indivs_df, families_df = parseFileToDFs(
"../gedcom_test_files/us33_list_orphans.ged")
orphans = validate.list_orphans(indivs_df, families_df)
expected = ['@lea@', '@luke@']
expected2 = ['@luke@', '@lea@']
self.assertTrue((orphans == expected) or (orphans == expected2))
def test(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/sprint1_acceptance_file.ged")
_, dates_indiv, _, _ = validate.dates_before_current_date(
indivs_df, fams_df)
expected_indiv = {'ID': {0: '@mystery@'}, 'DEATH': {0: '20 MAY 2200'}}
self.assertEqual(dates_indiv[['ID', 'DEATH']].to_dict(),
expected_indiv)
def test(self):
_, fams_df = parseFileToDFs(
"../gedcom_test_files/us39_list_upcoming_anniversaries.ged")
upcoming_aniversary_df = validate.list_upcoming_anniversaries(
fams_df, date(2018, 11, 18))
expected = {'HUSBAND NAME': {0: 'The /Force/'}}
self.assertEqual(expected,
upcoming_aniversary_df[['HUSBAND NAME']].to_dict())
def test(self):
indivs_df, families_df = parseFileToDFs(
"../gedcom_test_files/us13_sibling_spacing.ged")
strange_siblings = validate.siblings_spacing(indivs_df, families_df)
expected = {('@I1@', '@I4@', 6)}
expected2 = {('@I4@', '@I1@', 6)}
self.assertTrue(strange_siblings == expected
or strange_siblings == expected2)
def test_erroneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us12_parents_not_too_old_fail.ged")
indivs_error = validate.mother_too_old(indivs_df, fams_df)
expected = [['@shmi@']]
self.assertEqual(expected,
indivs_error[['ID_idv_mother']].values.tolist())
indivs_error = validate.father_too_old(indivs_df, fams_df)
expected = [['@mystery@']]
self.assertEqual(expected,
indivs_error[['ID_idv_father']].values.tolist())
def test_errorneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us28_order_siblings_by_age.ged")
ordered_fams_df = validate.order_siblings_by_age(indivs_df, fams_df)
self.assertEqual(
list(ordered_fams_df['CHILDREN'])[0][0], ('@ani4@', 'no birthday'))
self.assertEqual(
list(ordered_fams_df['CHILDREN'])[0][1],
('@ani@', '8401.0 days old'))
self.assertEqual(
list(ordered_fams_df['CHILDREN'])[0][2],
('@ani2@', '7672.0 days old'))
self.assertEqual(
list(ordered_fams_df['CHILDREN'])[0][3],
('@ani3@', '7671.0 days old'))
def test_erroneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us18_siblings_should_not_marry.ged")
wrong_roles = validate.siblings_should_not_marry(indivs_df, fams_df)
expected = [{
'ID_fam': '@solo@',
'ID_HUSBAND': '@luke@',
'ID_WIFE': '@lea@'
}]
actual = [
row.to_dict() for _, row in wrong_roles[
['ID_fam', 'ID_HUSBAND', 'ID_WIFE']].iterrows()
]
self.assertEqual(sorted(actual, key=lambda d: d['ID_fam']),
sorted(expected, key=lambda d: d['ID_fam']))
def test_errorneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us06_divorce_before_death.ged")
div_after_death = validate.divorce_before_death(indivs_df, fams_df)
expected = [{
'ID': '@shmi@',
'NAME': 'Shmi /Skywalker/',
'DEATH': '16 MAY 1977',
'DIVORCED': '10 MAY 1978'
}]
actual = [
row.to_dict() for _, row in div_after_death[
['ID', 'NAME', 'DEATH', 'DIVORCED']].iterrows()
]
self.assertEqual(sorted(actual, key=lambda dict: dict['ID']),
sorted(expected, key=lambda dict: dict['ID']))
def test(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/us38_list_upcoming_birthday.ged")
upcoming_birthday_df = validate.list_upcoming_birthday(
indivs_df, date(2018, 11, 18))
expected = [{
'ID': '@shmi1@',
'NAME': 'Shmiclone /Skywalker/',
'DAYS_TO_BIRTHDAY': 27
}]
actual = [
row.to_dict() for _, row in upcoming_birthday_df[
['ID', 'NAME', 'DAYS_TO_BIRTHDAY']].iterrows()
]
self.assertEqual(sorted(actual, key=lambda d: d['ID']),
sorted(expected, key=lambda d: d['ID']))
def test(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us08_birth_before_marriage.ged")
indivs_violation = validate.birth_before_parents_married(
indivs_df, fams_df)[['ID', 'NAME']]
expected = {
'ID': {
9: '@owen@',
18: '@luke@'
},
'NAME': {
9: 'Owen /Lars/',
18: 'Luke /Skywalker/'
}
}
self.assertEqual(indivs_violation.to_dict(), expected)
def test_erroneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us21_correct_gender_for_role.ged")
wrong_roles = validate.correct_gender_for_role(indivs_df, fams_df)
expected = [{
'ID': '@shmi@',
'GENDER': 'F'
}, {
'ID': '@mystery@',
'GENDER': 'M'
}]
actual = [
row.to_dict()
for _, row in wrong_roles[['ID', 'GENDER']].iterrows()
]
self.assertEqual(sorted(actual, key=lambda dict: dict['ID']),
sorted(expected, key=lambda dict: dict['ID']))
def test(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/sprint1_acceptance_file.ged")
marriage_after_death = validate.marriage_before_death(
indivs_df, fams_df)
expected = [{
'ID': '@shmi@',
'NAME': 'Shmi /Skywalker/',
'MARRIED': '20 MAY 1979',
'DEATH': '20 MAY 1976'
}]
actual = [
row.to_dict() for _, row in marriage_after_death[
['ID', 'NAME', 'MARRIED', 'DEATH']].iterrows()
]
self.assertEqual(sorted(actual, key=lambda dict: dict['ID']),
sorted(expected, key=lambda dict: dict['ID']))
def test_erroneous(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us25_unique_first_names.ged")
violations = validate.unique_first_names_in_families(
indivs_df, fams_df)
expected = {
'ID': {
6: '@ani2@'
},
'BIRTHDAY': {
6: '25 MAY 1977'
},
'NAME': {
6: 'Anakin /Skywalker/'
}
}
self.assertEqual(expected, violations[['ID', 'BIRTHDAY',
'NAME']].to_dict())
def test(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us37_list_recent_survivors.ged")
recent_survivors = validate.list_recent_survivors(indivs_df, fams_df)
expected = [{
'ID': '@shmi@',
'living spouses': {'@mystery@'},
'living descendants': {'@ani@', '@luke@', '@lea@', '@kylo@'}
}, {
'ID': '@ani@',
'living spouses': {'@padme@'},
'living descendants': {'@luke@', '@lea@', '@kylo@'}
}]
actual = [
row.to_dict() for _, row in recent_survivors[
['ID', 'living spouses', 'living descendants']].iterrows()
]
self.assertEqual(sorted(actual, key=lambda d: d['ID']),
sorted(expected, key=lambda d: d['ID']))
def test_erroneous(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/us31_list_single_older_than_30.ged")
violations = validate.list_living_single_older_than_30(indivs_df)
expected = {
'ID': {
6: '@buke@',
8: '@luke@'
},
'AGE': {
6: 52.0,
8: 41.0
},
'SPOUSE': {
6: None,
8: None
}
}
self.assertEqual(expected, violations[['ID', 'AGE',
'SPOUSE']].to_dict())
def test(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us09_birth_before_death_of_parents.ged")
violation_mother = validate.birth_before_parents_death_mother(
indivs_df, fams_df)[['ID_c', 'ID_m']]
violation_father = validate.birth_before_parents_death_father(
indivs_df, fams_df)[['ID_c', 'ID_m']]
expected_mother = {
'ID_c': {
2: '@luke@',
3: '@lea@'
},
'ID_m': {
2: '@padme@',
3: '@padme@'
}
}
expected_father = {'ID_c': {2: '@luke@'}, 'ID_m': {2: '@ani@'}}
self.assertEqual(violation_mother.to_dict(), expected_mother)
self.assertEqual(violation_father.to_dict(), expected_father)
def test(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/us10_married_before_14.ged")
violations = validate.marriage_before_14(indivs_df, fams_df)
expected = {
'ID': {
2: '@ani@',
3: '@han@'
},
'NAME': {
2: 'Anakin /Skywalker/',
3: 'Han /Solo/'
},
'AGE_MARRIED': {
2: 1,
3: 3
}
}
self.assertEqual(violations[['ID', 'NAME', 'AGE_MARRIED']].to_dict(),
expected)
def test_ALIVE(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/parser_test_ALIVE.ged")
expected = [{
'ID': '@mystery@',
'ALIVE': True
}, {
'ID': '@shmi@',
'ALIVE': False
}, {
'ID': '@owen@',
'ALIVE': True
}, {
'ID': '@ani@',
'ALIVE': False
}]
actual = [
row.to_dict() for _, row in indivs_df[['ID', 'ALIVE']].iterrows()
]
for act, exp in zip(sorted(actual, key=lambda dict: dict['ID']),
sorted(expected, key=lambda dict: dict['ID'])):
self.assertEqual(act, exp)
def test_get_descendents(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/utils_test_get_descendents.ged")
descendents = utils.get_descendants("@mystery@", fams_df)
expected = {'@ani@', '@luke@', '@lea@', '@kylo@'}
self.assertEqual(expected, descendents)
def test_erroneous(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/us23_unique_name_birthday.ged")
violations = validate.list_unique_name_birthday(indivs_df)
expected = {'NAME': {2: 'Shmi /Skywalker/'}}
self.assertEqual(expected, violations[['NAME']].to_dict())
def test_get_children(self):
indivs_df, fams_df = parseFileToDFs(
"../gedcom_test_files/utils_test_get_descendents.ged")
children = utils.get_children("@shmi@", fams_df)
expected = {'@owen@', '@ani@'}
self.assertEqual(expected, children)
def test(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/us32_list_multiple_births.ged")
multipleBirths = validate.multipleBirths(indivs_df)
expected = {'@luke@', '@lea@'}
self.assertEqual(expected, set(multipleBirths['ID'].values))
def test_erroneous(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/us36_list_recent_deaths.ged")
output = validate.list_recent_deaths(indivs_df)
expected = {'ID': {1: '@shmi@'}}
self.assertEqual(output[["ID"]].to_dict(), expected)
def test_erroneous(self):
indivs_df, _ = parseFileToDFs(
"../gedcom_test_files/us30_list_living_married.ged")
violations = validate.list_living_married(indivs_df)
expected = {'ID': {0: '@mystery@', 1: '@shmi@'}}
self.assertEqual(expected, violations[['ID']].to_dict())
