Function test_load_between_multiple_dimensions

tests/search/test_Route.py:682–699

Tests that the load_between() method correctly calculates the load for multiple dimensions.

(frm, to, dim, expected)

    ],
)
def test_load_between_multiple_dimensions(frm, to, dim, expected):
    """
    Tests that the load_between() method correctly calculates the load for
    multiple dimensions.
    """
    data = ProblemData(
        clients=[
            Client(1, 0, delivery=[1, 2]),
            Client(2, 0, delivery=[4, 5]),
        ],
        depots=[Depot(0, 0)],
        vehicle_types=[VehicleType(1, capacity=[10, 10])],
        distance_matrices=[np.zeros((3, 3), dtype=int)],
        duration_matrices=[np.zeros((3, 3), dtype=int)],
    )

    route = make_search_route(data, [1, 2])
    assert_equal(route.load_between(frm, to, dimension=dim).load(), expected)


def test_load_between_equal_to_before_after_when_one_is_depot_different_dims(