def test_empty(self): heartbeatGroups = [] routingTable = {} candidateSet = [] constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet) self.assertEqual(heartbeatGroups, [[]])
def test_three(self): heartbeatGroups = [] routingTable = {2: [3, 4], 3: [2, 4], 4:[3, 2]} candidateSet = [2, 3, 4] constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet) self.assertEqual(heartbeatGroups, [[2, 3, 4]])
def test_three_but_three_first(self): heartbeatGroups = [] routingTable = {2: [3, 4], 3: [2], 4:[2]} candidateSet = [3, 4, 2] constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet) self.assertEqual(heartbeatGroups, [[3, 2], [4]])
def test_two(self): heartbeatGroups = [] routingTable = {2: [3], 3: [2]} candidateSet = [2, 3] constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet) self.assertEqual(heartbeatGroups, [[2, 3]])