1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#include "testing/testing.hpp"
#include "routing/base/astar_algorithm.hpp"
#include "std/map.hpp"
#include "std/utility.hpp"
#include "std/vector.hpp"
namespace routing_test
{
using namespace routing;
struct Edge
{
Edge(unsigned v, double w) : v(v), w(w) {}
unsigned GetTarget() const { return v; }
double GetWeight() const { return w; }
unsigned v;
double w;
};
class UndirectedGraph
{
public:
using TVertexType = unsigned;
using TEdgeType = Edge;
void AddEdge(unsigned u, unsigned v, unsigned w)
{
m_adjs[u].push_back(Edge(v, w));
m_adjs[v].push_back(Edge(u, w));
}
void GetAdjacencyList(unsigned v, vector<Edge> & adj) const
{
adj.clear();
auto const it = m_adjs.find(v);
if (it != m_adjs.end())
adj = it->second;
}
void GetIngoingEdgesList(unsigned v, vector<Edge> & adj) const
{
GetAdjacencyList(v, adj);
}
void GetOutgoingEdgesList(unsigned v, vector<Edge> & adj) const
{
GetAdjacencyList(v, adj);
}
double HeuristicCostEstimate(unsigned v, unsigned w) const { return 0; }
private:
map<unsigned, vector<Edge>> m_adjs;
};
void TestAStar(UndirectedGraph const & graph, vector<unsigned> const & expectedRoute, double const & expectedDistance)
{
using TAlgorithm = AStarAlgorithm<UndirectedGraph>;
TAlgorithm algo;
RoutingResult<unsigned> actualRoute;
TEST_EQUAL(TAlgorithm::Result::OK, algo.FindPath(graph, 0u, 4u, actualRoute), ());
TEST_EQUAL(expectedRoute, actualRoute.path, ());
TEST_ALMOST_EQUAL_ULPS(expectedDistance, actualRoute.distance, ());
actualRoute.path.clear();
TEST_EQUAL(TAlgorithm::Result::OK, algo.FindPathBidirectional(graph, 0u, 4u, actualRoute), ());
TEST_EQUAL(expectedRoute, actualRoute.path, ());
TEST_ALMOST_EQUAL_ULPS(expectedDistance, actualRoute.distance, ());
}
UNIT_TEST(AStarAlgorithm_Sample)
{
UndirectedGraph graph;
// Inserts edges in a format: <source, target, weight>.
graph.AddEdge(0, 1, 10);
graph.AddEdge(1, 2, 5);
graph.AddEdge(2, 3, 5);
graph.AddEdge(2, 4, 10);
graph.AddEdge(3, 4, 3);
vector<unsigned> const expectedRoute = {0, 1, 2, 3, 4};
TestAStar(graph, expectedRoute, 23);
}
} // namespace routing_test
|