def test_server_response_copy(self): activity = models.Activity() activity.sort = "100" activity.times = "125,167,140,140,128,128,141,128,128,143,127,129,129,129,142,116,127,269" session = models_tests.generate_session(activities=[activity]) deleted, words_minute = main.get_filtered_times(session) self.assertEqual(deleted, 1) self.assertEqual(int(words_minute*1000), 449933)
def test_activities_sort(self): a = models.Activity() a.sort = "20" a.times = "1,2,3,2,1,2,3" a2 = models.Activity() a2.sort = "100" a2.times = "1,1,1,1,2,2,2,2,8,10" deleted, words_minute = main.get_filtered_times(models_tests.generate_session(activities=[a, a2])) self.assertEqual(deleted, 2) self.assertEqual(words_minute, 60*1000/1.5)
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)