def testHistTeamDraft(self): r1 = DeterministicRankingFunction(None, self.weights_1) r2 = DeterministicRankingFunction(None, self.weights_2) interleaved_list = [0, 1, 3, 2] htd = HistTeamDraft() self.assertEqual(htd.infer_outcome(interleaved_list, None, [0, 0, 0, 0], r1, r2, self.query), 0, "No clicks.") self.assertEqual(htd.infer_outcome(interleaved_list, None, [1, 0, 0, 0], r1, r2, self.query), 1, "Target rankers" " are the same as the original rankers, so ranker 2 has to win.") self.assertEqual(htd.infer_outcome(interleaved_list, None, [1, 0, 0, 0], r2, r1, self.query), -1, "Target rankers" " are switched, so ranker 1 has to win.")
def testHistTeamDraft(self): r1 = DeterministicRankingFunction(None, self.weights_1) r2 = DeterministicRankingFunction(None, self.weights_2) interleaved_list = [0, 1, 3, 2] htd = HistTeamDraft() self.assertEqual( htd.infer_outcome(interleaved_list, None, [0, 0, 0, 0], r1, r2, self.query), 0, "No clicks.") self.assertEqual( htd.infer_outcome(interleaved_list, None, [1, 0, 0, 0], r1, r2, self.query), 1, "Target rankers" " are the same as the original rankers, so ranker 2 has to win.") self.assertEqual( htd.infer_outcome(interleaved_list, None, [1, 0, 0, 0], r2, r1, self.query), -1, "Target rankers" " are switched, so ranker 1 has to win.")