Add default param constructors, create bool query for bfs and tests

This commit is contained in:
stefiosif
2022-09-09 18:12:38 +03:00
parent a4ddc3fbe7
commit 5c1b13b400
3 changed files with 161 additions and 157 deletions

View File

@@ -3,79 +3,21 @@
#include "algorithm/tarjan.h"
#include "algorithm/breadth_first_search.h"
#include <random>
#include <functional>
constexpr int verticesNum = 10;
#include <iostream>
using namespace graph;
TEST_SUITE("Algorithm") {
TEST_CASE("Tarjan::execute 1") {
// 1 --> 2 --> 4 --> 1
// 2 --> 3 --> 5 --> 7 --> 3
// 5 --> 9 --> 6 --> 8 --> 9
Digraph<std::uint16_t> G;
G.insert(1, 2);
G.insert(2, 3);
G.insert(2, 4);
G.insert(3, 5);
G.insert(4, 1);
G.insert(5, 7);
G.insert(5, 9);
G.insert(6, 8);
G.insert(7, 3);
G.insert(8, 9);
G.insert(9, 6);
REQUIRE_EQ(G.adjMatrix.size(), 9);
algo::Tarjan<std::uint16_t> tarjan(G);
auto r = tarjan.execute();
std::vector<std::vector<std::uint16_t>> x = {
{6, 8, 9},
{3, 5, 7},
{1, 2, 4}
};
for (auto i = 0; i < r.size(); i++) {
auto kv = std::views::keys(r[i].adjMatrix);
CHECK_EQ(std::is_permutation(kv.begin(), kv.end(), x[i].begin()), true);
}
}
TEST_CASE("Tarjan::execute 2") {
// 1 --> 2 --> 5 --> 7 --> 2
// 1 --> 4 --> 3 --> 1
// 4 --> 6 --> 3
Digraph<std::uint16_t> G;
G.insert(1, 2);
G.insert(1, 4);
G.insert(2, 5);
G.insert(3, 1);
G.insert(4, 3);
G.insert(4, 6);
G.insert(5, 7);
G.insert(6, 3);
G.insert(7, 2);
REQUIRE_EQ(G.adjMatrix.size(), 7);
algo::Tarjan<std::uint16_t> tarjan(G);
auto r = tarjan.execute();
std::vector<std::vector<std::uint16_t>> x = {
{2, 5, 7},
{1, 3, 4, 6}
};
for (auto i = 0; i < r.size(); i++) {
auto kv = std::views::keys(r[i].adjMatrix);
CHECK_EQ(std::is_permutation(kv.begin(), kv.end(), x[i].begin()), true);
}
}
TEST_CASE("Tarjan::execute 3") {
TEST_CASE("Tarjan::execute") {
// 1 --> 2 --> 3 --> 1
// 3 --> 4 --> 5 --> 3
// 2 --> 6 --> 7 --> 8 --> 6
// 7 --> 9
// 6 --> 9 --> 10 --> 11 --> 12 --> 13 --> 9
// 12 --> 10
Digraph<std::uint16_t> G;
@@ -100,53 +42,48 @@ TEST_SUITE("Algorithm") {
REQUIRE_EQ(G.adjMatrix.size(), 13);
algo::Tarjan<std::uint16_t> tarjan(G);
auto r = tarjan.execute();
auto SCCs = algo::Tarjan<std::uint16_t>(G.adjMatrix).execute();
std::vector<std::vector<std::uint16_t>> x = {
std::vector<std::vector<std::uint16_t>> expected = {
{9, 10, 11, 12, 13},
{6, 7, 8},
{1, 2, 3, 4, 5}
};
for (auto i = 0; i < r.size(); i++) {
auto kv = std::views::keys(r[i].adjMatrix);
CHECK_EQ(std::is_permutation(kv.begin(), kv.end(), x[i].begin()), true);
for (auto i = 0; i < SCCs.size(); i++) {
auto kv = std::views::keys(SCCs[i].adjMatrix);
CHECK_EQ(std::is_permutation(kv.begin(), kv.end(),
expected[i].begin()), true);
}
}
TEST_CASE("BreadthFirstSearch::execute 1") {
// 1 --> 2 --> 5 --> 7 --> 2
// 1 --> 4 --> 3 --> 1
// 4 --> 6 --> 3
TEST_CASE("Tarjan::execute ~ DAG") {
Digraph<std::uint16_t> G;
G.insert(1, 2);
G.insert(1, 4);
G.insert(2, 5);
G.insert(3, 1);
G.insert(4, 3);
G.insert(4, 6);
G.insert(5, 7);
G.insert(6, 3);
G.insert(7, 2);
auto gen = std::bind(std::uniform_real_distribution<>(0, 1), std::default_random_engine());
algo::BreadthFirstSearch<std::uint16_t> bfs(G);
auto r = bfs.execute(1);
for (int i = 0; i <= verticesNum; ++i) {
for (int j = i + 1; j <= verticesNum; ++j) {
if (gen() < 0.25) G.insert(i, j);
}
}
std::map<std::uint16_t, std::set<std::uint16_t>> x = {
{1, {2, 4}},
{2, {5}},
{4, {3, 6}},
{5, {7}}
};
auto SCCs = algo::Tarjan<std::uint16_t>(G.adjMatrix).execute();
CHECK_EQ(r, x);
// Testing whether an scc is a 1-vertex scc which after the normalization
// has no edges, in this case we know tgat there exist no 2-vertex sccs
for (auto& scc : SCCs) {
CHECK_EQ(std::all_of(scc.adjMatrix.begin(), scc.adjMatrix.end(),
[](const auto& p) {
return p.second.size() == 0;
}), true);
}
}
TEST_CASE("BreadthFirstSearch::execute 2") {
TEST_CASE("BreadthFirstSearch::execute") {
// 1 --> 2 --> 3 --> 1
// 3 --> 4 --> 5 --> 3
// 2 --> 6 --> 7 --> 8 --> 6
// 7 --> 9
// 6 --> 9 --> 10 --> 11 --> 12 --> 13 --> 9
// 12 --> 10
Digraph<std::uint16_t> G;
@@ -169,10 +106,10 @@ TEST_SUITE("Algorithm") {
G.insert(13, 9);
G.insert(12, 10);
algo::BreadthFirstSearch<std::uint16_t> bfs(G);
auto r = bfs.execute(1);
auto tree =
algo::BreadthFirstSearch<std::uint16_t>(G.adjMatrix).execute(1);
std::map<std::uint16_t, std::set<std::uint16_t>> x = {
std::map<std::uint16_t, std::set<std::uint16_t>> expected = {
{1, {2}},
{2, {3, 6}},
{3, {4}},
@@ -185,6 +122,41 @@ TEST_SUITE("Algorithm") {
{12, {13}}
};
CHECK_EQ(r, x);
CHECK_EQ(tree, expected);
}
TEST_CASE("BreadthFirstSearch::query") {
// 1 --> 2 --> 3 --> 1
// 3 --> 4 --> 5 --> 3
// 2 --> 6 --> 7 --> 8 --> 6
// 7 --> 9
// 6 --> 9 --> 10 --> 11 --> 12 --> 13 --> 9
// 12 --> 10
Digraph<std::uint16_t> G;
G.insert(1, 2);
G.insert(2, 3);
G.insert(3, 1);
G.insert(3, 4);
G.insert(4, 5);
G.insert(5, 3);
G.insert(2, 6);
G.insert(6, 7);
G.insert(7, 8);
G.insert(7, 9);
G.insert(8, 6);
G.insert(6, 9);
G.insert(9, 10);
G.insert(10, 11);
G.insert(11, 12);
G.insert(12, 13);
G.insert(13, 9);
G.insert(12, 10);
algo::BreadthFirstSearch<std::uint16_t> bfs(G.adjMatrix);
CHECK_EQ(bfs.query(1, 5), true);
CHECK_EQ(bfs.query(1, 10), true);
CHECK_EQ(bfs.query(9, 5), false);
CHECK_EQ(bfs.query(8, 4), false);
}
}