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())