This commit is contained in:
2022-04-26 18:39:15 +02:00
parent 358c1f386c
commit 5d07e043e5
2 changed files with 15 additions and 6 deletions

View File

@@ -1252,13 +1252,15 @@ class PairCheck:
self.score_b = report_b.score
for result_a, result_b in zip(report_a.lines, report_b.lines):
result = result_a["score"] - result_b["score"]
self._store_result(result, result_a["dataset"])
if result > 0:
self.winners.append(result_a["dataset"])
elif result < 0:
self.losers.append(result_a["dataset"])
else:
self.tie.append(result_a["dataset"])
def _store_result(self, result, dataset):
if result > 0:
self.winners.append(dataset)
elif result < 0:
self.losers.append(dataset)
else:
self.tie.append(dataset)
def report(self):
print(f"{'Model':<20} {'File':<70} {'Score':<10} Win Tie Lose")

View File

@@ -63,3 +63,10 @@ class PairCheckTest(unittest.TestCase):
) as f:
expected = f.read()
self.assertEqual(computed, expected)
def test_pair_check_store_result(self):
report = self.build_model(win=True, lose=True)
report.compute()
report._store_result(0, "balloons")
report._store_result(1, "balloons")
report._store_result(-1, "balloons")