class IntegrationTest(unittest.TestCase):
def setUp(self):
self.label_1 = "a"
self.label_2 = "b"
self.label_3 = "c"
self.true_positive = 8
self.true_negative = 8
self.false_positive = 8
self.false_negative = 8
self.ct = ConfusionTable(
self.label_1,
self.true_positive,
self.true_negative,
self.false_positive,
self.false_negative,
format_dict_as_grid
)
def test_generator_returns_confusion_table_correctly_with_two_classes(self):
predictions = pmap({
self.label_1: pmap({
self.label_1: self.true_positive,
self.label_2: self.false_positive,
}),
self.label_2: pmap({
self.label_1: self.false_negative,
self.label_2: self.true_negative
})
})
confusion_table = generate_confusion_table(predictions, self.label_1, format_dict_as_grid)
self.failUnlessEqual(self.ct.__str__(), confusion_table.__str__())
def test_generator_returns_confusion_table_correctly_with_three_classes(self):
predictions = pmap({
self.label_1: pmap({
self.label_1: self.true_positive,
self.label_2: self.false_positive / 2,
self.label_3: self.false_positive / 2
}),
self.label_2: pmap({
self.label_1: self.false_negative / 2,
self.label_2: self.true_negative / 2,
self.label_3: 0
}),
self.label_3: pmap({
self.label_1: self.false_negative / 2,
self.label_2: 0,
self.label_3: self.true_negative / 2
})
})
confusion_table = generate_confusion_table(predictions, self.label_1, format_dict_as_grid)
self.failUnlessEqual(self.ct.__str__(), confusion_table.__str__())
def setUp(self):
self.label_1 = "a"
self.label_2 = "b"
self.label_3 = "c"
self.true_positive = 8
self.true_negative = 8
self.false_positive = 8
self.false_negative = 8
self.ct = ConfusionTable(
self.label_1,
self.true_positive,
self.true_negative,
self.false_positive,
self.false_negative,
format_dict_as_grid
)
