def test_lastnight_records(self): rec1 = SleepRecord(datetime(2016, 1, 1, 0, 0), datetime(2016, 1, 1, 0, 15)) records = [ rec1 ] sut = SleepSet(records) sleeps_day = sut.get_nap_hrs() self.assertEqual(0, sleeps_day)
def test_daytime_records(self): rec1 = SleepRecord(datetime(2016, 1, 1, 7, 0), datetime(2016, 1, 1, 9, 0)) records = [ rec1 ] sut = SleepSet(records) sleeps_day = sut.get_nap_hrs() self.assertEqual(2, sleeps_day)
def test_naps(self): rec1 = SleepRecord(datetime(2016, 1, 1, 9, 0), datetime(2016, 1, 1, 9, 15)) rec2 = SleepRecord(datetime(2016, 1, 1, 9, 15), datetime(2016, 1, 1, 9, 30)) rec3 = SleepRecord(datetime(2016, 1, 1, 12, 0), datetime(2016, 1, 1, 12, 15)) rec4 = SleepRecord(datetime(2016, 1, 1, 12, 15), datetime(2016, 1, 1, 12, 30)) records = [ rec1, rec2, rec3, rec4 ] sut = SleepSet(records) self.assertEqual(2, sut.get_nap_count()) self.assertEqual(1.0, sut.get_nap_hrs())