def test_global(self):
models_session = [
models_tests.generate_model_session(),
models_tests.generate_model_session(),
models_tests.generate_model_session()
]
self.older_config_get_list_block_session_value = [
models_tests.generate_block_session(session=models_session[0], use_data=False),
models_tests.generate_block_session(session=models_session[1], use_data=False),
models_tests.generate_block_session(session=models_session[2], use_data=True)
]
self.older_config_get_current_block_Session_value = self.older_config_get_list_block_session_value
configuration = mocks.MockOlderConfig()
sessions = [
mocks.MockSession(model=models_session[0], status_begin=5,
completed_time=dateutil.parser.parse("2015-09-21")),
mocks.MockSession(model=models_session[1], status_begin=5,
completed_time=dateutil.parser.parse("2015-09-21")),
mocks.MockSession(model=models_tests.generate_model_session(),
status_begin=5, completed_time=dateutil.parser.parse("2015-09-21")),
mocks.MockSession(model=models_session[2], status_begin=5, completed_time=None),
mocks.MockSession(model=models_tests.generate_model_session(), completed_time=None),
]
list_sessions, sessions_made, sessions_use_data = main.generate_lists(configuration,
sessions)
self.assertEqual(list_sessions, models_session)
self.assertEqual(sessions_made, [sessions[0], sessions[1]])
self.assertEqual(sessions_use_data, [models_session[2]])
def setUp(self):
self.days = map(lambda e: dateutil.parser.parse("2015/09/" + str(e)), range(21, 28))
self.model_sessions = [
models_tests.generate_model_session(),
models_tests.generate_model_session(),
models_tests.generate_model_session()
]
def test(self):
list = [
mocks.MockSession(self.model_sessions[0], publish_date=None, completed_time=self.days[0]),
mocks.MockSession(self.model_sessions[0], publish_date=self.days[1], completed_time=self.days[2]),
mocks.MockSession(models_tests.generate_model_session(), publish_date=self.days[0],
completed_time=self.days[0]),
mocks.MockSession(self.model_sessions[0], publish_date=self.days[2], completed_time=self.days[2]),
mocks.MockSession(models_tests.generate_model_session(), publish_date=self.days[2],
completed_time=self.days[2]),
mocks.MockSession(self.model_sessions[1], publish_date=self.days[3], completed_time=self.days[3]),
mocks.MockSession(models_tests.generate_model_session(), publish_date=self.days[4], completed_time=None),
mocks.MockSession(self.model_sessions[2], publish_date=self.days[5], completed_time=None),
]
not_done, not_done_pattern, s_week = main.get_counters(list, self.model_sessions, self.days[2])
self.assertEqual(not_done, 2)
self.assertEqual(not_done_pattern, 1)
self.assertEqual(s_week, 3)
def test_with_data(self):
self.tmp = 0
def mock_get_percentile(older, session):
self.tmp += 1
return self.tmp
main.get_percentile = mock_get_percentile
ms1 = models_tests.generate_model_session(1)
ms2 = models_tests.generate_model_session(2)
ms3 = models_tests.generate_model_session(3)
sessions = [
models_tests.generate_session(model=ms1),
models_tests.generate_session(model=ms1),
models_tests.generate_session(model=ms1)
]
p, m = main.get_average_data(None, sessions, [ms2, ms3])
self.assertIsNone(p)
self.assertIsNone(m)
sessions = [
models_tests.generate_session(model=ms1, motivation=1),
models_tests.generate_session(model=ms2),
models_tests.generate_session(model=ms3)
]
p, m = main.get_average_data(None, sessions, [ms1])
self.assertEqual(p, 1)
self.assertEqual(m, 1)
sessions = [
models_tests.generate_session(model=ms1, motivation=1),
models_tests.generate_session(model=ms2, motivation=5),
models_tests.generate_session(model=ms3)
]
p, m = main.get_average_data(None, sessions, [ms1, ms2])
self.assertEqual(p, 2.5)
self.assertEqual(m, 3)