TestUtils.h

Go to the documentation of this file.

 
#ifndef _COMPAREIMAGES_H
#define _COMPAREIMAGES_H
 
#include "SEUtils/IsClose.h"
#include <boost/test/predicate_result.hpp>
 
namespace SourceXtractor {
 
template <typename T, typename U>
boost::test_tools::predicate_result compareImages(
  const T& ref, const U& val, double atol = 1e-8, double rtol = 1e-5) {
  boost::test_tools::predicate_result res(true);
 
  if (ref->getWidth() != val->getWidth() || ref->getHeight() != val->getHeight()) {
    res = false;
    res.message() << "Images do not match in size: "
                  << ref->getWidth() << 'x' << ref->getHeight() << " vs "
                  << val->getWidth() << 'x' << val->getHeight();
  }
 
  int w = ref->getWidth(), h = ref->getHeight();
  auto ref_chunk = ref->getChunk(0, 0, w, h);
  auto val_chunk = val->getChunk(0, 0, w, h);
 
  for (int x = 0; x < ref->getWidth(); ++x) {
    for (int y = 0; y < ref->getHeight(); ++y) {
      auto expected = ref_chunk->getValue(x, y);
      auto value = val_chunk->getValue(x, y);
      if (!isClose(expected, value, atol, rtol)) {
        res = false;
        res.message() << "Not matching values at position " << x << "," << y
                      << ": " << expected << " != " << value << "\n";
      }
    }
  }
 
  return res;
}
 
template <typename T, typename U>
boost::test_tools::predicate_result compareCollections(
  const T& ref, const U& val, double atol = 1e-8, double rtol = 1e-5) {
  boost::test_tools::predicate_result res(true);
 
  auto ref_i = std::begin(ref);
  auto val_i = std::begin(val);
  int i = 0;
 
  while (ref_i != std::end(ref) && val_i != std::end(val)) {
 
    if (!isClose(*ref_i, *val_i, atol, rtol)) {
      res = false;
      res.message() << "Not matching values at position " << i << ": " << *ref_i << " != " << *val_i << "\n";
    }
 
    ++ref_i;
    ++val_i;
    ++i;
  }
 
  if (ref_i != std::end(ref) || val_i != std::end(val)) {
    res = false;
    res.message() << "The sequences have different length!" << "\n";
  }
 
  return res;
}
 
boost::test_tools::predicate_result checkIsClose(double ref, const double val, double atol = 1e-8, double rtol = 1e-5) {
  boost::test_tools::predicate_result res(true);
  if (!isClose(ref, val, atol, rtol)) {
    res = false;
    res.message() << "Values not close enough: " << ref << " " << val << " with absolute tolerance " << atol
                  << " and relative " << rtol << "\n";
  }
  return res;
}
 
} // end SourceXtractor
 
#endif // _COMPAREIMAGES_H