def test_divorced_date(self): """add divorced date 1 APR 1970 """ divorced_date = datetime.strptime("1 APR 1970", '%d %b %Y') fam = Family("@F09@") fam.set_date("1 APR 1970", "divorced") self.assertEqual(divorced_date, fam.get_divorced_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())