arrays.hpp

Go to the documentation of this file.

// Copyright (c) 2015-2018 Commissariat à l'énergie atomique et aux énergies alternatives (CEA)
// Copyright (c) 2015-2018 Centre national de la recherche scientifique (CNRS)
// Copyright (c) 2018-2021 Simons Foundation
// Copyright (c) 2016 Igor Krivenko
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You may obtain a copy of the License at
//     https://www.gnu.org/licenses/gpl-3.0.txt
//
// Authors: Igor Krivenko, Olivier Parcollet, Nils Wentzell

 
#pragma once
 
#ifndef TRIQS_DEBUG
#define TRIQS_DEBUG
#endif
 
#include "../arrays.hpp"
 
#include <gtest/gtest.h>
#include <h5/h5.hpp>
#include <mpi/mpi.hpp>
#include <nda/nda.hpp>
#include <nda/gtest_tools.hpp>
 
#include <sstream>
#include <string>
 
using dcomplex = std::complex<double>;
using triqs::clef::placeholder;


#define EXPECT_PRINT(X, Y)                                                                                                                           \
  {                                                                                                                                                  \
    std::stringstream ss;                                                                                                                            \
    ss << (Y);                                                                                                                                       \
    EXPECT_EQ(X, ss.str());                                                                                                                          \
  }

#define ASSERT_PRINT(X, Y)                                                                                                                           \
  {                                                                                                                                                  \
    std::stringstream ss;                                                                                                                            \
    ss << (Y);                                                                                                                                       \
    ASSERT_EQ(X, ss.str());                                                                                                                          \
  }

#define MAKE_MAIN                                                                                                                                    \
  int main(int argc, char **argv) {                                                                                                                  \
    ::testing::InitGoogleTest(&argc, argv);                                                                                                          \
    if (mpi::has_env) {                                                                                                                              \
      mpi::environment env(argc, argv);                                                                                                              \
      std::cout << "MPI environment detected\n";                                                                                                     \
      return RUN_ALL_TESTS();                                                                                                                        \
    } else                                                                                                                                           \
      return RUN_ALL_TESTS();                                                                                                                        \
  }

template <typename T> T rw_h5(T const &x, std::string filename = "ess", std::string name = "x") {
  T y; // must be default constructible
  {
    h5::file file(filename + ".h5", 'w');
    h5_write(file, name, x);
  }
  {
    h5::file file(filename + ".h5", 'r');
    h5_read(file, name, y);
  }
  return y;
}