test_stats.cpp

#define BOOST_TEST_DYN_LINK
#include <boost/test/unit_test.hpp>

#include <chrono>
#include <thread>

#include "utils/Stats.hpp"
#include "mpi/MPISystem.hpp"
#include "utils/Types.hpp"
#include "test_helper.hpp"


void checkStats(int size){
    BOOST_REQUIRE(TestHelper::checkNumMPIProcsAvailable(size));
    BOOST_CHECK( size % 2 == 1);

    combigrid::CommunicatorType comm = TestHelper::getComm(size);

    if(comm == MPI_COMM_NULL){
        return;
    }

    int globalID;
    MPI_Comm_rank(MPI_COMM_WORLD, &globalID);

    combigrid::Stats::initialize();
    BOOST_TEST_CHECKPOINT("stats setAttribute");
    combigrid::Stats::setAttribute("group", std::to_string(globalID));

    BOOST_TEST_CHECKPOINT("stats wait start");
    combigrid::Stats::startEvent("wait 5 seconds");
    std::this_thread::sleep_for(std::chrono::microseconds{5000});
    combigrid::Stats::stopEvent("wait 5 seconds");
    combigrid::Stats::startEvent("wait 5 seconds");
    std::this_thread::sleep_for(std::chrono::microseconds{5000});
    combigrid::Stats::stopEvent("wait 5 seconds");

    BOOST_TEST_CHECKPOINT("stats partial");
    combigrid::Stats::startEvent("write partial twice");
    combigrid::Stats::writePartial("test_stats_output", comm);
    combigrid::Stats::writePartial("test_stats_output", comm);
    combigrid::Stats::stopEvent("write partial twice");
    BOOST_TEST_CHECKPOINT("stats partial again");
    combigrid::Stats::writePartial("test_stats_output", comm);

    BOOST_TEST_CHECKPOINT("stats finalize");
    combigrid::Stats::finalize();
    // MPI_Barrier(comm);
    BOOST_TEST_CHECKPOINT("stats write");
    combigrid::Stats::write("test_stats_output", comm);
    // MPI_Barrier(comm);
}

void testMeasureTime(int size, long for_milliseconds) {
  BOOST_REQUIRE(TestHelper::checkNumMPIProcsAvailable(size));

  combigrid::CommunicatorType comm = TestHelper::getComm(size);
  if (comm == MPI_COMM_NULL) {
    return;
  }

  combigrid::Stats::initialize();

  combigrid::Stats::Event e;
  std::chrono::milliseconds timespan(for_milliseconds); 
  std::this_thread::sleep_for(timespan);
  e.end = std::chrono::high_resolution_clock::now();

  long unsigned int dur_usec = combigrid::Stats::getEventDurationInUsec(e);

  BOOST_TEST(dur_usec >= for_milliseconds*1000);
  BOOST_TEST(dur_usec < 50*for_milliseconds*1000);

  combigrid::Stats::finalize();
}
BOOST_FIXTURE_TEST_SUITE(stats, TestHelper::BarrierAtEnd, *boost::unit_test::timeout(60))

BOOST_AUTO_TEST_CASE(test_3, * boost::unit_test::timeout(20)) {
    checkStats(1);
}

BOOST_AUTO_TEST_CASE(test_1, * boost::unit_test::timeout(20)) {
    checkStats(9);
}

BOOST_AUTO_TEST_CASE(test_time) {
  testMeasureTime(9,1);
  testMeasureTime(9,10);
  testMeasureTime(9,1000);
  MPI_Barrier(MPI_COMM_WORLD);
}

BOOST_AUTO_TEST_SUITE_END()