testlib.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <sstream>
#include <string>
#include <map>
#include <sys/stat.h>

#include <TPRegexp.h>
#include <TString.h>

#include <cmath>

#include <TVector3.h>

using namespace std;

static unsigned int iTest = 1;
static unsigned int gPrecision = 12;
static unsigned int nTests = 0;
static double eps = 1.e-10;

static unsigned int gCurToDoMarker = 0;
class ToDo;
static std::map<int, ToDo*> gToDoMarkers;

#include "testlib.h"

int GetPrecision() { return gPrecision; }
int GetFieldWidth() { return gPrecision + 3; }

static Planner gPlanner;

Planner::~Planner()
{
  if (fDoPrint) {
    cout << "1.." << iTest-1 << endl;
  }
}

void
plan(int n)
{
  nTests = n;
  if (nTests != 0) {
    cout << "1.." << nTests << endl;
  }
  else {
    gPlanner.SetDoPrint();
  }
}

void
noplan()
{
  gPlanner.SetDoPrint();
}

void
diag(const std::string& msg)
{
  TString s(msg.c_str());

  TPRegexp repl("^");
  repl.Substitute(s, "# ", "mg");
  TPRegexp repl2("\\n$");
  repl.Substitute(s, "", "");
  cout << s.Data() << endl;
}

void
diag(const std::ostringstream& msg)
{
  diag(msg.str());
}

void
pass(const std::string& name)
{
  cout << "ok " << iTest++ << todoMsg(name) << endl;
}

void
fail(const std::string& name)
{
  cout << "not ok " << iTest++ << todoMsg(name) << endl;
}

bool
ok(bool okay)
{
  bool retval = true;
  if (!okay) {
    cout << "not ";
    retval = false;
  }
  cout << "ok " << iTest++  << todoMsg() << endl;
  return retval;
}

bool
ok(bool okay, const std::string& name)
{
  bool retval = true;
  if (!okay) {
    cout << "not ";
    retval = false;
  }
  cout << "ok " << iTest++ << todoMsg(name) << endl;
  return retval;
}

bool
is(const double a, const double b)
{
  return is(a, b, string(""));
}

bool
is(const double a, const double b, const std::string& name)
{
  bool retval = true;
  const double thisEps = eps;
  if (!( SAMENONFINITE(a,b) || (a+thisEps > b && a-thisEps < b) )) {
    cout << "not ";
    retval = false;
  }
  cout << "ok " << iTest++ << todoMsg(name);
  if (retval == false) {
    ostringstream s;
    s << ", Difference: test="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << a
      << " vs. ref="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << b;
    diag(s);
  }
  else
     cout << endl;
  return retval;
}

bool
isnot(const double a, const double b)
{
  return isnot(a, b, string(""));
}

bool
isnot(const double a, const double b, const std::string& name)
{
  bool retval = true;
  if (BOTHINFINITY(a,b) || BOTHNAN(a,b) || ((a+eps > b) && (a-eps < b))) {
    cout << "not ";
    retval = false;
  }
  cout << "ok " << iTest++ << todoMsg(name);
  if (retval == false) {
    ostringstream s;
    s << ", Unexpectedly equivalent: test="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << a
      << " vs. ref="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << b;
    diag(s);
  }
  else
     cout << endl;
  return retval;
}



bool
is(const float a, const float b)
{ return is((double)a, (double)b); }

bool
is(const float a, const float b, const std::string& name)
{ return is((double)a, (double)b, name); }

bool
isnot(const float a, const float b)
{ return isnot((double)a, (double)b); }

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

bool
isEps(const double a, const double b, const double thisEps, const std::string& name)
{
  bool retval = true;
  if (!( SAMENONFINITE(a,b) || (a+thisEps > b && a-thisEps < b) )) {
    cout << "not ";
    retval = false;
  }
  cout << "ok " << iTest++ << todoMsg(name);
  if (retval == false) {
    ostringstream s;
    s << ", Difference: test="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << a
      << " vs. ref="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << b;
    diag(s);
  }
  else
     cout << endl;
  return retval;
}

bool
isEpsDiag(const double a, const double b, const double thisEps, const std::string& name)
{
  bool retval = true;
  if (!( SAMENONFINITE(a,b) || (a+thisEps > b && a-thisEps < b) )) {
    cout << "not ";
    retval = false;
  }
  cout << "ok " << iTest++ << todoMsg(name);
  if (retval == false) {
    ostringstream s;
    s << ", Difference: test="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << a
      << " vs. ref="
      << setprecision(GetPrecision()) << setw(GetFieldWidth()) << b;
    diag(s);
  }
  else
     cout << endl;
  return retval;
}

bool
isDiag(const double a, const double b, const std::string& name)
{ return isEpsDiag(a, b, eps, name); }

bool
isDiag(const double a, const double b)
{ return isEpsDiag(a, b, eps, string("")); }

bool
isRelEpsDiag(const double a, const double b, const double thisEps, const std::string& name)
{
  return isEpsDiag(a, b, (fabs(a+b) == 0. ? 1.e-12 : fabs(a+b)*thisEps/2.), name);
}

bool
isDiag(const std::vector<double>& a, const std::vector<double>& b, const std::string& name)
{
  return isEpsDiag(a, b, eps, name);
}

bool
isEpsDiag(const std::vector<double>& a, const std::vector<double>& b, const double thisEps, 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 (!( SAMENONFINITE(a[i],b[i]) || (a[i]+thisEps > b[i] && a[i]-thisEps < 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
     cout << endl;
  return retval;
}

bool
isRelEpsDiag(const std::vector<double>& a, const std::vector<double>& b, const double thisEps, 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) {
    const double teps = fabs(a[i]+b[i]) == 0. ? 1.e-12 : fabs(a[i]+b[i])*thisEps/2.;
    if (!( SAMENONFINITE(a[i],b[i]) || (a[i]+teps > b[i] && a[i]-teps < 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
     cout << endl;
  return retval;
}


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

  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 (!( SAMENONFINITE(a[i],b[i]) || (a[i]+thisEps > b[i] && a[i]-thisEps < b[i]) )) {
      retval = false;
      break;
    }
  }

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


std::string
todoMsg(const std::string& name)
{
  if (gToDoMarkers.empty()) {
    if (name.length() == 0)
      return string("");
    return string(" - ") + name;
  }
  return string(" # TODO & SKIP ") + gToDoMarkers.rbegin()->second->GetReason()
          + string(" (testname: ") + name + string(")");
}

std::string
todoMsg()
{
  return todoMsg(string(""));
}

ToDo::ToDo(const std::string reason)
  : fReason(reason)
{
  fMarker = gCurToDoMarker++;
  gToDoMarkers[fMarker] = this;
}

ToDo::~ToDo()
{
  gToDoMarkers.erase(fMarker);
}


bool
fexists(const std::string& filename)
{
  bool okay = false;
  struct stat buffer;
  if (stat(filename.c_str(), &buffer)) {
    cout << "not ";
  }
  else
    okay = true;
  
  cout << "ok " << iTest++ << todoMsg(string("File ") + filename + string(" exists")) << endl;
  return okay;
}

unsigned int&
GetTestNo()
{
  return iTest;
}

bool
isEpsDiag(const TVector3& a, const TVector3& b, const double thisEps, const std::string& name)
{
  bool retval = fabs(a.x() - b.x()) < thisEps && fabs(a.y() - b.y()) < thisEps && fabs(a.z() - b.z()) < thisEps;
  if (!retval) {
     std::cout << "(" << a.x() << ", " << a.y() << ", " << a.z() << ") != "
               << "(" << b.x() << ", " << b.y() << ", " << b.z() << "), "
       << "not ";
    retval = false;
  }
  std::cout << "ok " << GetTestNo()++ << todoMsg(name);

  if (!retval) {
    ostringstream s;
    s << ", Diference (" << a.x() - b.x() << ", " << a.y() - b.y() << ", " << a.z() -  b.z() << ") != ";
    diag(s);
    retval = false;
  }
  else
     cout << endl;

  return retval;
}