def test_not_in_graph(self):
matrix = [[7, 4, 3], [3, 1, 2], [3, 0, 0]]
h = Hungarian(matrix)
h._init_labels()
self.assertTrue(h._in_equality_graph(0, 0))
self.assertTrue(h._in_equality_graph(1, 0))
self.assertTrue(h._in_equality_graph(2, 0))
def test_in_graph(self):
matrix = [[7, 4, 3], [3, 1, 2], [3, 0, 0]]
h = Hungarian(matrix)
h._init_labels()
self.assertFalse(h._in_equality_graph(0, 1))
self.assertFalse(h._in_equality_graph(1, 1))
self.assertFalse(h._in_equality_graph(2, 1))
self.assertFalse(h._in_equality_graph(0, 2))
self.assertFalse(h._in_equality_graph(1, 2))
self.assertFalse(h._in_equality_graph(2, 2))