def test_married_date(self):
"""add married date 28 MAR 2001
"""
married_date = datetime.strptime("28 MAR 2001", '%d %b %Y')
fam = Family("@F01@")
fam.set_date("28 MAR 2001", "married")
self.assertEqual(married_date, fam.get_married_date())
def test_both_dates(self):
"""add both dates together
"""
fam = Family("@F01@")
married_date = datetime.strptime("28 MAR 2001", '%d %b %Y')
fam.set_date("28 MAR 2001", "married")
divorced_date = datetime.strptime("1 APR 1970", '%d %b %Y')
fam.set_date("1 APR 1970", "divorced")
self.assertEqual(married_date, fam.get_married_date())
self.assertEqual(divorced_date, fam.get_divorced_date())
