from unittest import TestCase
from alex.ml.hypothesis import ConfusionNetwork
[docs]class TestConfusionNetwork(TestCase):
[docs] def test_iter(self):
dacn = ConfusionNetwork()
dacn.add(0.2, 1)
dacn.add(0.7, 2)
dacn.add(0.1, 3)
lst = list(dacn)
self.assertTrue(lst[0][0] == 0.2)
self.assertTrue(lst[0][1] == 1)
self.assertTrue(lst[1][0] == 0.7)
self.assertTrue(lst[1][1] == 2)
self.assertTrue(lst[2][0] == 0.1)
self.assertTrue(lst[2][1] == 3)
[docs] def test_remove(self):
dacn = ConfusionNetwork()
dacn.add(0.2, 1)
dacn.add(0.7, 2)
dacn.add(0.1, 3)
dacn.remove(1)
dacn.remove(3)
self.assertTrue(2 in dacn)
self.assertTrue(len(dacn) == 1)