testlib.h

Go to the documentation of this file.

#ifndef _testlib_h_
#define _testlib_h_

#include <iostream>
#include <iomanip>
#include <sstream>
#include <string>
#include <vector>
#include <map>

class TVector3;

extern unsigned int iTest;
extern double eps;

int GetPrecision();
int GetFieldWidth();

void plan(int n);
void noplan();
std::string todoMsg(const std::string& name);
std::string todoMsg();
unsigned int& GetTestNo();

void diag(const std::string& msg);
void diag(const std::ostringstream& msg);

void pass(const std::string& name);
void fail(const std::string& name);

bool ok(bool okay);
bool ok(bool okay, const std::string& name);

bool is(const double a, const double b);
bool is(const double a, const double b, const std::string& name);
bool is(const float a, const float b);
bool is(const float a, const float b, const std::string& name);
bool isDiag(const double a, const double b, const std::string& name);
bool isDiag(const double a, const double b);
bool isEpsDiag(const TVector3& a, const TVector3& b, const double thisEps, const std::string& name);

bool isEps(const double a, const double b, const double thisEps, const std::string& name);
bool isEpsDiag(const double a, const double b, const double thisEps, const std::string& name);
bool isRelEpsDiag(const double a, const double b, const double thisEps, const std::string& name);

bool isnot(const double a, const double b);
bool isnot(const double a, const double b, const std::string& name);
bool isnot(const float a, const float b);
bool isnot(const float a, const float b, const std::string& name);

#define BOTHFINITE(a,b) (isfinite(a) && isfinite(b))
#define BOTHNAN(a,b) (std::isnan(a) && std::isnan(b))
#define BOTHINFINITY(a,b) (isinf(a) == isinf(b) && isinf(a))
#define BOTHFINITE_OREQ(a,b) (BOTHNAN(a,b) || BOTHINFINITY(a,b) || BOTHFINITE(a,b))
#define SAMENONFINITE(a,b) (BOTHNAN(a,b) || BOTHINFINITY(a,b))

template<class T>
bool
is(const T& a, const T& b)
{
  return is(a, b, std::string(""));
}

template<class T>
bool
is(const T& a, const T& b, const std::string& name)
{
  bool retval = true;
  if (!(a == b)) {
    std::cout << "not ";
    retval = false;
  }
  std::cout << "ok " << GetTestNo()++ << todoMsg(name) << std::endl;
  return retval;
}


template<class T>
bool
isDiag(const T& a, const T& b)
{ return isDiag(a, b, std::string("")); }

template<class T>
bool
isDiag(const T& a, const T& b, const std::string& name)
{
  bool retval = true;
  if (!(a == b)) {
    std::cout << "not ";
    retval = false;
  }
  std::cout << "ok " << GetTestNo()++ << todoMsg(name);
  if (retval == false) {
    std::ostringstream s;
    //s << "Difference: test='" << a << "' vs. ref='" << b << "'";
    s << ", Difference: test="
      << std::setprecision(GetPrecision()) << std::setw(GetFieldWidth()) << a
      << " vs. ref="
      << std::setprecision(GetPrecision()) << std::setw(GetFieldWidth()) << b;
    diag(s);
  }
  else
     std::cout << std::endl;
  return retval;
}


template<class T>
bool
is(const std::vector<T>& a, const std::vector<T>& b, const std::string& name)
{
  bool retval = true;
  const unsigned int na = a.size();
  const unsigned int nb = b.size();

  std::string nName = name + std::string(" (vector size)");
  if (!is(na, nb, nName))
    return false;

  unsigned int i = 0;
  for (i = 0; i < na; ++i) {
    if (!(a[i] == b[i])) {
      retval = false;
      break;
    }
  }

  if (retval == false) {
    std::cout << "not ";
  }
  std::cout << "ok " << GetTestNo()++ << todoMsg(name) << std::endl;
  return retval;
}

template<class T>
bool
isDiag(const std::vector<T>& a, const std::vector<T>& b, const std::string& name)
{
  bool retval = true;
  const unsigned int na = a.size();
  const unsigned int nb = b.size();

  std::string nName = name + std::string(" (vector size)");
  if (!isDiag(na, nb, nName))
    return false;

  unsigned int i = 0;
  for (i = 0; i < na; ++i) {
    if (!(a[i] == b[i])) {
      retval = false;
      break;
    }
  }

  if (retval == false) {
    std::cout << "not ";
  }
  std::cout << "ok " << GetTestNo()++ << todoMsg(name);

  if (retval == false) {
    std::ostringstream s;
    //s << "Difference at index " << i << ": test='" << a[i] << "' vs. ref='" << b[i] << "'";
    s << "Difference at index " << i
      << ": test='"
      << std::setprecision(GetPrecision()) << std::setw(GetFieldWidth()) << a[i]
      << "' vs. ref='"
      << std::setprecision(GetPrecision()) << std::setw(GetFieldWidth()) << b[i]
      << "'";
    diag(s);
  }
  else
     std::cout << std::endl;
  return retval;
}


bool is(const std::vector<double>& a, const std::vector<double>& b, const std::string& name);
bool isDiag(const std::vector<double>& a, const std::vector<double>& b, const std::string& name);
bool isEpsDiag(const std::vector<double>& a, const std::vector<double>& b, const double thisEps, const std::string& name);
bool isRelEpsDiag(const std::vector<double>& a, const std::vector<double>& b, const double thisEps, const std::string& name);

template<class T>
bool
isnot(const T& a, const T& b)
{
  return isnot(a, b, std::string(""));
}

template<class T>
bool
isnot(const T& a, const T& b, const std::string name)
{
  bool retval = true;
  if (a == b) {
    std::cout << "not ";
    retval = false;
  }
  std::cout << "ok " << GetTestNo()++ << todoMsg(name) << std::endl;
  return retval;
}

template<class T>
bool
isnotDiag(const T& a, const T& b)
{
  return isnotDiag(a, b, std::string(""));
}

template<class T>
bool
isnotDiag(const T& a, const T& b, const std::string& name)
{
  bool retval = true;
  if (a == b) {
    std::cout << "not ";
    retval = false;
  }
  std::cout << "ok " << GetTestNo()++ << todoMsg(name);
  if (retval == false) {
    std::ostringstream s;
    s << "Unexpectectly equivalent: test='"
      << std::setprecision(GetPrecision()) << std::setw(GetFieldWidth()) << a
      << "' vs. ref='"
      << std::setprecision(GetPrecision()) << std::setw(GetFieldWidth()) << b << "'";
    diag(s);
  }
  else
     std::cout << std::endl;
  return retval;
}


class ToDo {
public:
  ToDo(const std::string reason);
  ~ToDo();
  std::string GetReason() { return fReason; }
private:
  std::string fReason;
  int fMarker;
};


class Planner {
public:
  Planner() : fDoPrint(false) {}
  ~Planner();
  void SetDoPrint() { fDoPrint = true; }
private:
  bool fDoPrint;
};

bool fexists(const std::string& filename);

#endif