Function test_distance_calculation

tests/test_Solution.py:409–433

Tests that route distance calculations are correct, and that the overall Solution's distance is the sum of the route distances.

(ok_small)

def test_distance_calculation(ok_small):
    """
    Tests that route distance calculations are correct, and that the overall
    Solution's distance is the sum of the route distances.
    """
    sol = Solution(ok_small, [[1, 2], [3], [4]])
    routes = sol.routes()

    # Solution is feasible, so all its routes should also be feasible.
    assert_(sol.is_feasible())
    assert_(all(route.is_feasible() for route in routes))

    # Solution distance should be equal to all routes' distances. These we
    # check separately.
    assert_equal(sol.distance(), sum(route.distance() for route in routes))

    distances = ok_small.distance_matrix(profile=0)
    expected = distances[0, 1] + distances[1, 2] + distances[2, 0]
    assert_equal(routes[0].distance(), expected)

    expected = distances[0, 3] + distances[3, 0]
    assert_equal(routes[1].distance(), expected)

    expected = distances[0, 4] + distances[4, 0]
    assert_equal(routes[2].distance(), expected)


def test_excess_load_calculation(ok_small):