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)
def test_no_data(self): p, m = main.get_average_data(None, [], []) self.assertIsNone(p) self.assertIsNone(m)