#include <doctest/doctest.h>

#include "graph/scc.h"
#include "graph/digraph.h"
#include "algorithm/tarjan.h"

#include <vector>

using namespace graph;

TEST_SUITE("Graph") {

  TEST_CASE("Reverse") {
    // 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.vertices.size(), 9);

    auto RG = G.reverse();

    std::map<std::uint16_t, std::set<std::uint16_t>> exp = {
      {1, {4}},
      {2, {1}},
      {3, {2, 7}},
      {4, {2}},
      {5, {3}},
      {6, {9}},
      {7, {5}},
      {8, {6}},
      {9, {5, 8}}
    };
    
    CHECK_EQ(RG.adjMatrix, exp);

  }

  TEST_CASE("SCC T1") {
    // 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.vertices.size(), 9);

    algo::Tarjan<std::uint16_t> tarjan(G);
    auto sccs = tarjan.findSCC();
  }

  TEST_CASE("SCC T2") {
    // 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.vertices.size(), 7);

    algo::Tarjan<std::uint16_t> tarjan(G);
    auto sccs = tarjan.findSCC();
  }
}