def test_new(self): a = DayTally(self.a_date) self.assertEqual(a.date(), self.a_date) expected = Node.list_of(self.tt_day_begin) self.assertSameTallies(a.get_tallies(), expected)
def test_add_first(self): a = DayTally(self.a_date) a.add(self.first_begin, self.first_end) expected = Node.list_of(self.tt_day_begin, self._ttally_of(self.first_begin, 1), self._ttally_of(self.first_end)) self.assertSameTallies(a.get_tallies(), expected)
def test_add_right_same(self): a = DayTally(self.a_date) second_begin = datetime.combine(self.a_date, time(12, 30)) a.add(self.first_begin, self.first_end) a.add(second_begin, self.first_end) expected = Node.list_of(self.tt_day_begin, self._ttally_of(self.first_begin, 1), self._ttally_of(second_begin, 2), self._ttally_of(self.first_end, 0)) self.assertSameTallies(a.get_tallies(), expected)
def test_add_several(self): a = DayTally(self.a_date) second_begin = datetime.combine(self.a_date, time(10, 0)) second_end = datetime.combine(self.a_date, time(11, 45)) third_begin = datetime.combine(self.a_date, time(12, 45)) third_end = datetime.combine(self.a_date, time(14, 0)) fourth_begin = datetime.combine(self.a_date, time(10, 00)) fourth_end = datetime.combine(self.a_date, time(14, 0)) a.add(self.first_begin, self.first_end) a.add(second_begin, second_end) a.add(third_begin, third_end) a.add(fourth_begin, fourth_end) expected = [ self.tt_day_begin, self._ttally_of(second_begin, 2), self._ttally_of(second_end, 1), self._ttally_of(self.first_begin, 2), self._ttally_of(third_begin, 3), self._ttally_of(self.first_end, 2), self._ttally_of(third_end, 0) ] expected_nodes = Node.list_of(*expected) self.assertSameTallies(a.get_tallies(), expected_nodes) self.assertEquals(a.get_tallies().as_list(), expected)