def test_age_stupid_raises_error(self): date_of_birth = date(2010, 1, 1) user = User("id", date_of_birth) with self.assertRaises(ValueError): user.age(date(2000, 1, 1))
def test_age(self): date_of_birth = date(2012, 1, 2) user = User("id", date_of_birth) self.assertEqual(user.age(date(2016, 1, 1)), 3) self.assertEqual(user.age(date(2016, 1, 2)), 4)