osrm-backend/unit_tests/contractor/graph_contractor.cpp
2022-08-24 18:13:49 +02:00

134 lines
4.2 KiB
C++

#include "contractor/graph_contractor.hpp"
#include "../common/range_tools.hpp"
#include "helper.hpp"
#include <boost/test/unit_test.hpp>
#include <tbb/global_control.h>
using namespace osrm;
using namespace osrm::contractor;
using namespace osrm::unit_test;
BOOST_AUTO_TEST_SUITE(graph_contractor)
BOOST_AUTO_TEST_CASE(contract_graph)
{
tbb::global_control scheduler(tbb::global_control::max_allowed_parallelism, 1);
/*
* <--1--<
* (0) >--3--> (1) >--3--> (3)
* v ^ v ^
* \ / \ |
* 1 1 1 1
* \ ^ \ /
* >(5) > (4) >
*/
std::vector<TestEdge> edges = {TestEdge{0, 1, 3},
TestEdge{0, 5, 1},
TestEdge{1, 3, 3},
TestEdge{1, 4, 1},
TestEdge{3, 1, 1},
TestEdge{4, 3, 1},
TestEdge{5, 1, 1}};
auto reference_graph = makeGraph(edges);
auto contracted_graph = reference_graph;
std::vector<bool> core = contractGraph(contracted_graph, {1, 1, 1, 1, 1, 1});
// This contraction order is dependent on the priority caculation in the contractor
// but deterministic for the same graph.
CHECK_EQUAL_RANGE(core, false, false, false, false, false, false);
/* After contracting 0 and 2:
*
* Deltes edges 5 -> 0, 1 -> 0
*
* <--1--<
* (0) ---3--> (1) >--3--> (3)
* \ ^ v ^
* \ / \ |
* 1 1 1 1
* \ ^ \ /
* >(5) > (4) >
*/
reference_graph.DeleteEdgesTo(5, 0);
reference_graph.DeleteEdgesTo(1, 0);
/* After contracting 5:
*
* Deletes edges 1 -> 5
*
* <--1--<
* (0) ---3--> (1) >--3--> (3)
* \ ^ v ^
* \ / \ |
* 1 1 1 1
* \ / \ /
* >(5) > (4) >
*/
reference_graph.DeleteEdgesTo(5, 0);
reference_graph.DeleteEdgesTo(1, 0);
/* After contracting 3:
*
* Deletes edges 1 -> 3
* Deletes edges 4 -> 3
* Insert edge 4 -> 1
*
* <--1---
* (0) ---3--> (1) >--3--- (3)
* \ ^ v ^ |
* \ / \ \ |
* 1 1 1 2 1
* \ / \ \ /
* >(5) > (4) >
*/
reference_graph.DeleteEdgesTo(1, 3);
reference_graph.DeleteEdgesTo(4, 3);
// Insert shortcut
reference_graph.InsertEdge(4, 1, {2, 4, 1.0, 3, 0, true, true, false});
/* After contracting 4:
*
* Delete edges 1 -> 4
*
* <--1---
* (0) ---3--> (1) >--3--- (3)
* \ ^ v ^ |
* \ / \ \ |
* 1 1 1 2 1
* \ / \ \ /
* >(5) \ (4) >
*/
reference_graph.DeleteEdgesTo(1, 4);
/* After contracting 1:
*
* Delete no edges.
*
* <--1---
* (0) ---3--> (1) >--3--- (3)
* \ ^ v ^ |
* \ / \ \ |
* 1 1 1 2 1
* \ / \ \ /
* >(5) \ (4) >
*/
REQUIRE_SIZE_RANGE(contracted_graph.GetAdjacentEdgeRange(0), 2);
BOOST_CHECK(contracted_graph.FindEdge(0, 1) != SPECIAL_EDGEID);
BOOST_CHECK(contracted_graph.FindEdge(0, 5) != SPECIAL_EDGEID);
REQUIRE_SIZE_RANGE(contracted_graph.GetAdjacentEdgeRange(1), 0);
REQUIRE_SIZE_RANGE(contracted_graph.GetAdjacentEdgeRange(2), 0);
REQUIRE_SIZE_RANGE(contracted_graph.GetAdjacentEdgeRange(3), 3);
BOOST_CHECK(contracted_graph.FindEdge(3, 1) != SPECIAL_EDGEID);
BOOST_CHECK(contracted_graph.FindEdge(3, 4) != SPECIAL_EDGEID);
REQUIRE_SIZE_RANGE(contracted_graph.GetAdjacentEdgeRange(4), 2);
BOOST_CHECK(contracted_graph.FindEdge(4, 1) != SPECIAL_EDGEID);
REQUIRE_SIZE_RANGE(contracted_graph.GetAdjacentEdgeRange(5), 1);
BOOST_CHECK(contracted_graph.FindEdge(5, 1) != SPECIAL_EDGEID);
}
BOOST_AUTO_TEST_SUITE_END()