#include <doctest/extensions/doctest_mpi.h>

int f_for_test(int rank);

int f_for_test(int rank) {
    if (rank == 0) {
        return 10;
    } else if (rank == 1) {
        return 11;
    }
    return 0;
}

MPI_TEST_CASE("Parallel test on 2 processes", 2) { // if MPI_SIZE < 2, report test can't be run
    // 3 objects accessible in the test:
    //    test_comm: MPI_Comm of size 2
    //    test_rank: integer of value the rank of the process in test_comm
    //    test_nb_procs: integer of value the size of the process (here: 2)

    int x = f_for_test(test_rank);

    MPI_CHECK(0, x == 10); // CHECK for rank 0, that x==10
    MPI_CHECK(1, x == 11); // CHECK for rank 1, that x==11
                           // MPI_CHECK( 2,  x==0 ); // will trigger a static assert because non-existing rank
}

MPI_TEST_CASE("Parallel test on 3 processes (failing)", 3) {
    int x = f_for_test(test_rank);

    MPI_CHECK(0, x == 10); // CHECK for rank 0, that x==10
    MPI_CHECK(1, x == 11); // CHECK for rank 1, that x==11
    MPI_CHECK(2, x == -1); // CHECK for rank 2, that x==-1 (which is not the case -> will trigger a failure report)
}

MPI_TEST_CASE("Parallel tests with subcases", 2) {

    SUBCASE("sub_case 0") {
        CHECK(test_nb_procs == 2);
    }
    SUBCASE("sub_case 1") {
        CHECK(test_rank == 0); // should fail on proc 1
    }
}
