def test_properties(self):
c = ConfusionMatrix([[1, 2], [3, 4]])
cn = c.normalise()
self.assertEqual(c.num_classes, 2)
self.assertEqual(c.total, 10)
self.assertEqual(cn.total, 1)
def test_normalise(self):
c = ConfusionMatrix([[1, 2], [3, 4]])
cn = c.normalise()
self.assertTrue(
np.all(cn.asarray() == np.array([[0.1, 0.2], [0.3, 0.4]])))