def test_non_dtm(self): score = MagicMock() score.mate = None score.cp = 100 info_handler = MagicMock() info_handler.info = {'score': [None, score]} self.assertEqual(annotator.eval_human(True, info_handler), "1.00") self.assertEqual(annotator.eval_human(False, info_handler), "-1.00")
def test_negative_dtm(self): score = MagicMock() score.mate = -5 score.cp = None info_handler = MagicMock() info_handler.info = {'score': [None, score]} self.assertEqual(annotator.eval_human(True, info_handler), "Mate in 5") self.assertEqual(annotator.eval_human(False, info_handler), "Mate in 5")