def test_evaluate(self):
node_profile = NodeProfileMultiObjective(dep_times=[3, 1], label_class=LabelTime)
node_profile.update([LabelTime(departure_time=3, arrival_time_target=4)])
self.assertEquals(4, min_arrival_time_target(node_profile.evaluate(3)))
node_profile.update([LabelTime(departure_time=1, arrival_time_target=1)])
self.assertEquals(1, min_arrival_time_target(node_profile.evaluate(1)))
def test_identity_profile(self):
identity_profile = NodeProfileMultiObjective(dep_times=[10])
identity_profile.update([LabelTimeWithBoardingsCount(10, 10, 0, True)])
self.assertEqual(10, min_arrival_time_target(identity_profile.evaluate(10, first_leg_can_be_walk=True)))