class TestDateHelper(unittest.TestCase):
def setUp(self):
self.dh = DateHelper()
self.date_str = "1999-08-01"
self.datetime_str = "1998-05-02 14:08:00"
self.date_obj = self.dh.make_date(self.date_str)
self.datetime_obj = self.dh.make_datetime(self.datetime_str)
def test_print_date(self):
expected = "01 Aug 1999"
actual = self.dh.print_date(self.date_obj)
self.assertEqual(expected, actual)
def test_print_datetime(self):
expected = "14:08:00 02 May 1998"
actual = self.dh.print_datetime(self.datetime_obj)
self.assertEqual(expected, actual)
def test_print_nice(self):
expected = "2:08pm, 2 May 1998"
actual = self.dh.print_nice(self.datetime_obj)
self.assertEqual(expected, actual)
def print_recovery(self):
dh = DateHelper()
recovery_time = self.start + self.duration + self.recovery
return dh.print_nice(recovery_time)
def print_duration(self):
dh = DateHelper()
treatment_time = self.start + self.duration
return dh.print_nice(treatment_time)
def print_start(self):
dh = DateHelper()
start_time = dh.print_nice(self.start)
return start_time