#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#ifdef HAVE_MPI
#include <mpi.h>
#endif

#ifndef BOOST_TEST_NO_MAIN
#define BOOST_TEST_NO_MAIN
#endif

#ifdef HAVE_BOOST_UNIT_TEST_FRAMEWORK_LIB
#define BOOST_TEST_DYN_LINK
#include <boost/test/unit_test.hpp>
#else
#include <boost/test/included/unit_test.hpp>
#endif

#include <boost/test/framework.hpp>
#include <boost/test/detail/unit_test_parameters.hpp>
#include <boost/test/impl/unit_test_main.ipp>

#ifdef BOOST_TEST_ALTERNATIVE_INIT_API
bool init_unit_test()
#else
::boost::unit_test::test_suite* init_unit_test_suite(int argc, char* argv[])
#endif
{
  boost::unit_test::framework::master_test_suite().p_name.value = "VMG Test Suite";
#ifdef BOOST_TEST_ALTERNATIVE_INIT_API
  return true;
#else
  return NULL;
#endif
}

int BOOST_TEST_CALL_DECL
main(int argc, char* argv[])
{
#ifdef HAVE_MPI
  MPI_Init(&argc, &argv);
#endif

  // prototype for user's unit test init function
#ifdef BOOST_TEST_ALTERNATIVE_INIT_API
  boost::unit_test::init_unit_test_func init_func = &init_unit_test;
#else
  boost::unit_test::init_unit_test_func init_func = &init_unit_test_suite;
#endif

  int rval = ::boost::unit_test::unit_test_main(init_func, argc, argv);

#ifdef HAVE_MPI
  MPI_Finalize();
#endif

  return rval;
}