def test_whenConfusionMatrixAreMergedCountsAreAdded(self):
rewardClasses = [0,1]
confusionMatrix1 = ConfusionMatrix(rewardClasses)
confusionMatrix2 = ConfusionMatrix(rewardClasses)
realReward = 0
predictedReward = 0
confusionMatrix2.addObservation(realReward, predictedReward)
confusionMatrix1.merge(confusionMatrix2)
self.assertEqual(1, confusionMatrix1.confusionMatrix[realReward][predictedReward])
