def test_ReadPredictionsCorruptedEmptyKey(self): t = temp_file_with_contents([ ',N\n', ]) with self.assertRaises(SystemExit) as context: evaluator.read_predictions(t) os.remove(t) self.assertEqual(context.exception.code, evaluator.EXIT_STATUS_PREDICTIONS_MALFORMED)
def test_ReadPredictionsMissingColumn(self): t = temp_file_with_contents([ 'P1,E\n', '"P2"\n', ]) with self.assertRaises(SystemExit) as context: evaluator.read_predictions(t) os.remove(t) self.assertEqual(context.exception.code, evaluator.EXIT_STATUS_PREDICTIONS_MALFORMED)
def test_ReadPredictions(self): t = temp_file_with_contents([ 'P1,E\n', '"P2",N\n', ]) predictions = evaluator.read_predictions(t) os.remove(t) self.assertEqual(predictions, { "P1": "E", "P2": "N", })
def test_ReadPredictions(self): t = temp_file_with_contents([ 'Q1,A\n', '"Q2",A;B\n', 'Q3,"A;B;C"\n', ]) predictions = evaluator.read_predictions(t) os.remove(t) self.assertEqual(predictions, { "Q1": ["A"], "Q2": ["A", "B"], "Q3": ["A", "B", "C"], })