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)