mat_checks.h

Go to the documentation of this file.

#ifndef MAT_CHECKS_H
#define MAT_CHECKS_H

#include "../bdm/itpp_ext.h"
#include "UnitTest++.h"

namespace UnitTest
{

bool AreClose(const itpp::vec &expected, const itpp::vec &actual,
              double tolerance);

bool AreClose(const itpp::vec &expected, const itpp::vec &actual,
              const itpp::vec &tolerance);

bool AreClose(const itpp::mat &expected, const itpp::mat &actual,
              double tolerance);

inline void CheckClose(TestResults &results, const itpp::vec &expected,
                       const itpp::vec &actual, double tolerance,
                       TestDetails const &details) {
    if (!AreClose(expected, actual, tolerance)) { 
        MemoryOutStream stream;
        stream << "Expected " << expected << " +/- " << tolerance << " but was " << actual;

        results.OnTestFailure(details, stream.GetText());
    }
}

inline void CheckClose(TestResults &results, const itpp::mat &expected,
                       const itpp::mat &actual, double tolerance,
                       TestDetails const &details) {
    if (!AreClose(expected, actual, tolerance)) { 
        MemoryOutStream stream;
        stream << "Expected " << expected << " +/- " << tolerance << " but was " << actual;

        results.OnTestFailure(details, stream.GetText());
    }
}

}

class CurrentContext
{
private:
    static const char *config_name;
    static int index;

public:
    // the pointer must stay valid for the lifetime of the object
    CurrentContext(const char *name, int idx);
    ~CurrentContext();

    template<typename Expected, typename Actual>
    static void CheckEqualEx(UnitTest::TestResults& results,
                             Expected const& expected,
                             Actual const& actual,
                             UnitTest::TestDetails const& details) {
        if (!(expected == actual)) {
            UnitTest::MemoryOutStream stream;
            stream << "error at " << config_name << '[' << index << "]: expected " << expected << " but was " << actual;

            results.OnTestFailure(details, stream.GetText());
        }
    }

    template<typename Expected, typename Actual, typename Tolerance>
    static void CheckCloseEx(UnitTest::TestResults& results,
                             Expected const& expected,
                             Actual const& actual,
                             Tolerance const& tolerance,
                             UnitTest::TestDetails const& details) {
        if (!UnitTest::AreClose(expected, actual, tolerance)) { 
            UnitTest::MemoryOutStream stream;
            stream << "error at " << config_name << '[' << index << "]: expected " << expected << " +/- " << tolerance << " but was " << actual;

            results.OnTestFailure(details, stream.GetText());
        }
    }
};

#define CHECK_EQUAL_EX(expected, actual) \
    do \
    { \
        try { \
            CurrentContext::CheckEqualEx(*UnitTest::CurrentTest::Results(), expected, actual, UnitTest::TestDetails(*UnitTest::CurrentTest::Details(), 0, false)); \
        } \
        catch (...) { \
            UnitTest::CurrentTest::Results()->OnTestFailure(UnitTest::TestDetails(*UnitTest::CurrentTest::Details(), __LINE__), \
                    "Unhandled exception in CHECK_EQUAL_EX(" #expected ", " #actual ")"); \
        } \
    } while (0)


#define CHECK_CLOSE_EX(expected, actual, tolerance) \
    do \
    { \
        try { \
            CurrentContext::CheckCloseEx(*UnitTest::CurrentTest::Results(), expected, actual, tolerance, UnitTest::TestDetails(*UnitTest::CurrentTest::Details(), 0, false)); \
        } \
        catch (...) { \
            UnitTest::CurrentTest::Results()->OnTestFailure(UnitTest::TestDetails(*UnitTest::CurrentTest::Details(), __LINE__), \
                    "Unhandled exception in CHECK_CLOSE_EX(" #expected ", " #actual ")"); \
        } \
    } while (0)

#endif

---