Function test_problem_data_replace_with_changes

tests/test_ProblemData.py:325–366

Tests that when calling ``ProblemData.replace()`` indeed replaces the data values with those passed to the method.

()

def test_problem_data_replace_with_changes():
    """
    Tests that when calling ``ProblemData.replace()`` indeed replaces the
    data values with those passed to the method.
    """
    clients = [Client(x=0, y=0, delivery=[0])]
    depots = [Depot(x=0, y=0)]
    vehicle_types = [VehicleType(2, capacity=[1])]
    mat = np.zeros((2, 2), dtype=int)
    original = ProblemData(clients, depots, vehicle_types, [mat], [mat])

    # Let's replace the clients, vehicle types, and the distance matrix, each
    # with different values than in the original data. The duration matrix
    # is left unchanged.
    new = original.replace(
        clients=[Client(x=1, y=1, delivery=[0])],
        vehicle_types=[VehicleType(3, [4]), VehicleType(5, [6])],
        distance_matrices=[np.where(np.eye(2), 0, 2)],
    )

    assert_(new is not original)
    assert_(new.location(1) is not original.location(1))
    assert_(new.location(1).x != original.location(1).x)
    assert_(new.location(1).y != original.location(1).y)

    for idx in range(original.num_vehicle_types):  # only compare first type
        new_veh_type = new.vehicle_type(idx)
        orig_veh_type = original.vehicle_type(idx)

        assert_(new_veh_type is not orig_veh_type)
        assert_(new_veh_type.capacity != orig_veh_type.capacity)
        assert_(new_veh_type.num_available != orig_veh_type.num_available)

    assert_(new.distance_matrix(0) is not original.distance_matrix(0))
    with assert_raises(AssertionError):
        assert_equal(new.distance_matrix(0), original.distance_matrix(0))

    assert_(new.duration_matrix(0) is not original.duration_matrix(0))
    assert_equal(new.duration_matrix(0), original.duration_matrix(0))

    assert_equal(new.num_clients, original.num_clients)
    assert_(new.num_vehicle_types != original.num_vehicle_types)


def test_problem_data_replace_raises_mismatched_argument_shapes():