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)))