def test_found_free_y(self):
matrix = [[7, 4, 3], [3, 1, 2], [3, 0, 0]]
h = Hungarian(matrix)
h._init_labels()
h.matching = {}
h.inverse_matching = {}
t = h._find_augmenting_path({0: None}, {0}, set())
self.assertEqual(t, (0, 0, {0: None}))
def test_augment_matching(self):
matrix = [[1, 6, 0], [0, 8, 6], [4, 0, 1]]
h = Hungarian(matrix)
h._init_labels()
h.matching = {1: 1, 2: 0}
h.inverse_matching = {0: 2, 1: 1}
path = {1: 0, 0: None}
h._augment_matching(1, 2, path)
self.assertEqual(h.matching, {0: 1, 1: 2, 2: 0})
self.assertEqual(h.inverse_matching, {1: 0, 2: 1, 0: 2})
def test_neighbourhood_s_equals_t(self):
matrix = [[7, 4, 3], [3, 1, 2], [3, 0, 0]]
h = Hungarian(matrix)
h._init_labels()
h.matching = {0: 0}
h.inverse_matching = {0: 0}
h._update_labels = MagicMock()
copy_find_augmenting_path = h._find_augmenting_path
h._find_augmenting_path = MagicMock()
copy_find_augmenting_path({1: None, 0: 1}, {1, 0}, {0})
h._update_labels.assert_called_once_with({1, 0}, {0})
def test_found_matched_y(self):
matrix = [[7, 4, 3], [3, 1, 2], [3, 0, 0]]
h = Hungarian(matrix)
h._init_labels()
h.matching = {0: 0}
h.inverse_matching = {0: 0}
copy_find_augmenting_path = h._find_augmenting_path
h._find_augmenting_path = MagicMock()
copy_find_augmenting_path({1: None}, {1}, set())
h._find_augmenting_path.assert_called_once_with({
1: None,
0: 1
}, {1, 0}, {0})