mirror of
https://github.com/Doctorado-ML/benchmark.git
synced 2025-08-15 23:45:54 +00:00
Fix test
This commit is contained in:
@@ -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")
|
||||
|
@@ -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")
|
||||
|
Reference in New Issue
Block a user