def test_get_placement_of_subvertex(self):
"""
checks the placements get placement method
:return:
"""
subv = list()
for i in range(5):
subv.append(PartitionedVertex(None, ""))
pl = list()
for i in range(4):
pl.append(Placement(subv[i], 0, 0, i))
pls = Placements(pl)
for i in range(4):
self.assertEqual(pls.get_placement_of_subvertex(subv[i]), pl[i])
def test_get_subvertex_on_processor(self):
"""
checks that from a placements object, you can get to the correct
subvertex using the get_subvertex_on_processor() method
:return:
"""
subv = list()
for i in range(5):
subv.append(PartitionedVertex(None, ""))
pl = list()
for i in range(4):
pl.append(Placement(subv[i], 0, 0, i))
pls = Placements(pl)
for i in range(4):
self.assertEqual(pls.get_subvertex_on_processor(0, 0, i), subv[i])
self.assertEqual(pls.get_placement_of_subvertex(subv[0]), pl[0])
