def test_mapping_argmax_last(self): m = self.ensemble.approximate() m.outputs[0].lower = 0 m.outputs[0].upper = 0 m.outputs[1].lower = 1 m.outputs[1].upper = 1 m.outputs[2].lower = 2 m.outputs[2].upper = 2 self.assertEqual(2, vote.mapping_argmax(m))
def test_mapping_argmax_unknown(self): m = self.ensemble.approximate() self.assertEqual(-1, vote.mapping_argmax(m))