def test_AllTP(self):
npTrue = np.array([[[1],[1]],[[1],[1]]])
npPred = np.array([[[1],[1]],[[1],[1]]])
npColourMap = Error.ColouredPredMap(npPred, npTrue)
npExpectedColourMap = np.array([[self.npTP, self.npTP],
[self.npTP, self.npTP]])
self.assertEqual(npColourMap[0].tolist(), npExpectedColourMap.tolist())
def test_3Ch(self):
npTrue = np.array([[[0,0,1],[1,0,0]],[[0,0,1],[0,1,0]]])
npPred = np.array([[[1],[1]],[[1],[1]]])
npColourMap = Error.ColouredPredMap(npPred, npTrue[:,:,1:2])
npExpectedColourMap = np.array([[self.npFP, self.npFP],
[self.npFP, self.npTP]])
self.assertEqual(npColourMap[0].tolist(), npExpectedColourMap.tolist())
def test_GenCase(self):
npTrue = np.array([[[0],[1]],[[1],[0]]])
npPred = np.array([[[0],[1]],[[0],[1]]])
npColourMap = Error.ColouredPredMap(npPred, npTrue)
npExpectedColourMap = np.array([[self.npTN, self.npTP],
[self.npFN, self.npFP]], dtype= np.uint8)
self.assertEqual(npColourMap[0].tolist(), npExpectedColourMap.tolist())