def test_from_events_invalid(self):
"""Tests from_events() with invalid situations."""
# No start.
events = [Event(0, False)]
expected = [TimePeriod(None, 0)]
self.assertEqual(
expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# No end.
events = [Event(0, True)]
expected = [TimePeriod(0, None)]
self.assertEqual(
expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Multiple start.
events = [Event(0, True), Event(1, True), Event(2, False)]
expected = [TimePeriod(0, 2)]
self.assertEqual(
expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Multiple end.
events = [Event(0, True), Event(1, False), Event(2, False)]
expected = [TimePeriod(0, 1)]
self.assertEqual(
expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
def test_from_events_invalid(self):
"""Tests from_events() with invalid situations."""
# No start.
events = [Event(0, False)]
expected = [TimePeriod(None, 0)]
self.assertEqual(expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# No end.
events = [Event(0, True)]
expected = [TimePeriod(0, None)]
self.assertEqual(expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Multiple start.
events = [Event(0, True), Event(1, True), Event(2, False)]
expected = [TimePeriod(0, 2)]
self.assertEqual(expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Multiple end.
events = [Event(0, True), Event(1, False), Event(2, False)]
expected = [TimePeriod(0, 1)]
self.assertEqual(expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
def test_from_events(self):
"""Tests from_events()."""
# No periods.
events = []
expected = []
self.assertEqual(
expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Standard period.
events = [Event(0, True), Event(1, False)]
expected = [TimePeriod(0, 1)]
self.assertEqual(
expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Multiple periods.
events = [
Event(0, True),
Event(1, False),
Event(2, True),
Event(3, False)
]
expected = [TimePeriod(0, 1), TimePeriod(2, 3)]
self.assertEqual(
expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
def test_from_events(self):
"""Tests from_events()."""
# No periods.
events = []
expected = []
self.assertEqual(expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Standard period.
events = [Event(0, True), Event(1, False)]
expected = [TimePeriod(0, 1)]
self.assertEqual(expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
# Multiple periods.
events = [Event(0, True), Event(1, False), Event(2, True),
Event(3, False)]
expected = [TimePeriod(0, 1), TimePeriod(2, 3)]
self.assertEqual(expected,
TimePeriod.from_events(events, self.is_start_func,
self.is_end_func))
