def test_yyyy_to_age(self):
mock_today = datetime.date(2017, 1, 29)
with patch('datetime.date', Mock(today=lambda: mock_today)):
for sample_birthday in [19490301, '19490301', '194903', 194903]:
age = inputs.yyyy_to_age(sample_birthday)
self.assertEqual(age, 68)
def test_yyyy_to_age_none(self):
for sample_birthday in ['', float('nan')]:
age = inputs.yyyy_to_age(sample_birthday)
self.assertEqual(age, inputs.UNKNOWN)
def test_yyyy_to_age(self):
for sample_birthday in [19490301, '19490301', '194903', 194903]:
age = inputs.yyyy_to_age(sample_birthday)
self.assertEqual(age, 68)