def test_sort_simple(self): gt = [0, 1, 1] est = [0, 1, 1] cm = ConfMatrix(2, labelset = self.makelabelset(2)) cm.add(gt, est) self.assertEqual(cm.cm[1, 1], 2) self.assertEqual(cm.labelset[0], 'a') cm.sort() self.assertEqual(cm.cm[0, 0], 2) self.assertEqual(cm.labelset[0], 'b')
def test_sort_more_complicated(self): gt = [0, 0, 0, 1, 1, 1, 1, 1, 2, 3, 3] est = [0, 0, 0, 1, 1, 1, 0, 0, 2, 3, 3] cm = ConfMatrix(4, labelset = self.makelabelset(4)) cm.add(gt, est) self.assertEqual(cm.cm[1, 1], 3) self.assertEqual(cm.cm[3, 3], 2) self.assertEqual(cm.cm[1, 0], 2) self.assertEqual(cm.labelset, ['a', 'b', 'c', 'd']) cm.sort() self.assertEqual(cm.cm[0, 0], 3) self.assertEqual(cm.cm[2, 2], 2) self.assertEqual(cm.cm[0, 1], 2) self.assertEqual(cm.labelset, ['b', 'a', 'd', 'c'])