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'])
