def test_add_periods_from_extras_even_if_it_matches_discard_pattern(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(user_id=3,
question_index=4,
time=START_TIME,
event_type='get task')
create_question_event(user_id=3,
question_index=4,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task')
compute_task_periods(discard_periods=({
'user_id': 3,
'task_index': 4
}, ),
extra_periods=({
'user_id':
3,
'task_index':
4,
'start':
datetime.datetime(2000, 1, 1, 12, 0, 1, 0),
'end':
datetime.datetime(2000, 1, 1, 12, 0, 2, 0),
}, ))
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
def test_add_periods_from_extras_even_if_it_matches_discard_pattern(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(
user_id=3,
question_index=4,
time=START_TIME,
event_type='get task'
)
create_question_event(
user_id=3,
question_index=4,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task'
)
compute_task_periods(
discard_periods=({'user_id': 3, 'task_index': 4},),
extra_periods=({
'user_id': 3,
'task_index': 4,
'start': datetime.datetime(2000, 1, 1, 12, 0, 1, 0),
'end': datetime.datetime(2000, 1, 1, 12, 0, 2, 0),
},))
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
def test_make_tasks_with_events_interleaved_between_users(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(user_id=0, time=START_TIME, event_type='get task')
create_question_event(
user_id=1,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='get task'
)
create_question_event(
user_id=0,
time=START_TIME + datetime.timedelta(seconds=2),
event_type='post task'
)
create_question_event(
user_id=1,
time=START_TIME + datetime.timedelta(seconds=3),
event_type='post task'
)
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 2)
user_ids = [task_period.user_id for task_period in TaskPeriod.select()]
self.assertIn(0, user_ids)
self.assertIn(1, user_ids)
def test_make_task_only_if_both_events_from_one_user(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(user_id=0,
time=START_TIME,
event_type='get task')
create_question_event(user_id=1,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task')
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 0)
def test_skip_task_period_with_unmatching_task_indexes(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(question_index=0,
time=START_TIME,
event_type='get task')
create_question_event(question_index=1,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task')
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 0)
def test_make_task_only_if_both_events_from_one_user(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(user_id=0, time=START_TIME, event_type='get task')
create_question_event(
user_id=1,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task'
)
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 0)
def test_skip_task_period_with_unmatching_task_indexes(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(question_index=0, time=START_TIME, event_type='get task')
create_question_event(
question_index=1,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task'
)
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 0)
def test_add_periods_with_extras_specification(self):
compute_task_periods(extra_periods=({
'user_id': 3,
'task_index': 4,
'start': datetime.datetime(2000, 1, 1, 12, 0, 1, 0),
'end': datetime.datetime(2000, 1, 1, 12, 0, 2, 0),
},))
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
period = task_periods[0]
self.assertEqual(period.user_id, 3)
self.assertEqual(period.task_index, 4)
self.assertEqual(period.start, datetime.datetime(2000, 1, 1, 12, 0, 1, 0))
self.assertEqual(period.end, datetime.datetime(2000, 1, 1, 12, 0, 2, 0))
def test_add_periods_with_extras_specification(self):
compute_task_periods(
extra_periods=({
'user_id': 3,
'task_index': 4,
'start': datetime.datetime(2000, 1, 1, 12, 0, 1, 0),
'end': datetime.datetime(2000, 1, 1, 12, 0, 2, 0),
}, ))
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
period = task_periods[0]
self.assertEqual(period.user_id, 3)
self.assertEqual(period.task_index, 4)
self.assertEqual(period.start,
datetime.datetime(2000, 1, 1, 12, 0, 1, 0))
self.assertEqual(period.end, datetime.datetime(2000, 1, 1, 12, 0, 2,
0))
def test_make_task_period(self):
# Setup: create two events bounding a single task
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(time=START_TIME, event_type='get task')
create_question_event(time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task')
# Test: make sure a task has been created with the time bounds of the
# events that started and ended it.
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
period = task_periods[0]
self.assertEqual(period.user_id, 0)
self.assertEqual(period.start,
datetime.datetime(2000, 1, 1, 12, 0, 1, 0))
self.assertEqual(period.end, datetime.datetime(2000, 1, 1, 12, 0, 2,
0))
def test_make_task_period(self):
# Setup: create two events bounding a single task
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(time=START_TIME, event_type='get task')
create_question_event(
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task'
)
# Test: make sure a task has been created with the time bounds of the
# events that started and ended it.
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
period = task_periods[0]
self.assertEqual(period.user_id, 0)
self.assertEqual(period.start, datetime.datetime(2000, 1, 1, 12, 0, 1, 0))
self.assertEqual(period.end, datetime.datetime(2000, 1, 1, 12, 0, 2, 0))
def test_dont_discard_periods_that_dont_match_discard_pattern(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(user_id=3,
question_index=5,
time=START_TIME,
event_type='get task')
create_question_event(user_id=3,
question_index=5,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task')
compute_task_periods(
discard_periods=({
'user_id': 3,
'question_index': 4
}, ),
extra_periods=(),
)
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
def test_dont_discard_periods_that_dont_match_discard_pattern(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(
user_id=3,
question_index=5,
time=START_TIME,
event_type='get task'
)
create_question_event(
user_id=3,
question_index=5,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='post task'
)
compute_task_periods(
discard_periods=({'user_id': 3, 'question_index': 4},),
extra_periods=(),
)
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 1)
def test_make_tasks_with_events_interleaved_between_users(self):
START_TIME = datetime.datetime(2000, 1, 1, 12, 0, 1, 0)
create_question_event(user_id=0,
time=START_TIME,
event_type='get task')
create_question_event(user_id=1,
time=START_TIME + datetime.timedelta(seconds=1),
event_type='get task')
create_question_event(user_id=0,
time=START_TIME + datetime.timedelta(seconds=2),
event_type='post task')
create_question_event(user_id=1,
time=START_TIME + datetime.timedelta(seconds=3),
event_type='post task')
compute_task_periods(extra_periods=())
task_periods = TaskPeriod.select()
self.assertEqual(task_periods.count(), 2)
user_ids = [task_period.user_id for task_period in TaskPeriod.select()]
self.assertIn(0, user_ids)
self.assertIn(1, user_ids)