def test_addCycleSuccessors(self):
"""
Generates a list of nodes, appends it to a test node and checks, if
add_successor was called for every node.
:return:
"""
test_node = CycleNode(float(Fraction(0)))
test_node.add_cycle_successor = MagicMock()
node_list = [(Node(), float(Fraction(0))) for _ in range(10)]
test_node.add_cycle_successors(node_list)
for elem in node_list:
self.assertIn((elem,), test_node.add_cycle_successor.call_args_list)
def test_addCycleSuccessorsOfWhichOneAlreadyExists(self):
"""
Generates a list of nodes and duplicates some of them.
Then checks, if appending the list to a node raises an Exception.
:return:
"""
test_node = CycleNode(float(Fraction(0)))
duplicate_node = CycleNode(float(Fraction(0)))
node_list = [(duplicate_node, float(Fraction(0)))]
node_list.extend(
[(CycleNode(float(Fraction(0))), float(Fraction(0))) for _ in range(10)]
)
node_list.append((duplicate_node, float(Fraction(0))))
with self.assertRaises(Exception):
test_node.add_cycle_successors(node_list)
