LossTests.hpp

Go to the documentation of this file.

#ifndef LOSSTESTS_HPP_
#define LOSSTESTS_HPP_


#include <gtest/gtest.h>

// Redefine word "public" so every class field/method will be accessible for tests.
#define private public
#include "Loss.hpp"


class Vectors4x1Float : public ::testing::Test {
public:
    // Constructor. Sets layer size.
    Vectors4x1Float () {
        target_y = MAKE_MATRIX_PTR(float, 4, 1);
        predicted_y = MAKE_MATRIX_PTR(float, 4, 1);
    }

protected:
    // Sets values.
    virtual void SetUp() {
        (*target_y)  << 2.0, 3.0, 4.0, 5.0;
        (*predicted_y) << 1.0, 2.0, 3.0, 4.0;
    }

private:
    // Target y values.
    mic::types::MatrixPtr<float> target_y;

    // Predicted y values.
    mic::types::MatrixPtr<float> predicted_y;
};

class Vectors4x1Float2 : public ::testing::Test {
public:
    // Constructor. Sets layer size.
    Vectors4x1Float2 () {
        target_y = MAKE_MATRIX_PTR(float, 4,1);
        predicted_y1 = MAKE_MATRIX_PTR(float, 4, 1);
        predicted_y2 = MAKE_MATRIX_PTR(float, 4, 1);
    }

protected:
    // Sets values.
    virtual void SetUp() {
        (*target_y) << .4, .1, .25, .25;
        (*predicted_y1) << .25, .25, .25, .25;
        (*predicted_y2) << .4, .1, .1, .4;
    }

private:
    // Target y values.
    mic::types::MatrixPtr<float> target_y;

    // Predicted y values 1.
    mic::types::MatrixPtr<float> predicted_y1;
    // Predicted y values 2.
    mic::types::MatrixPtr<float> predicted_y2;
};


class Vectors3x2Float : public ::testing::Test {
public:
    // Constructor. Sets layer size.
    Vectors3x2Float () {
        target_y = MAKE_MATRIX_PTR(float, 3, 2);
        predicted_y = MAKE_MATRIX_PTR(float, 3, 2);
    }

protected:
    // Sets values.
    virtual void SetUp() {
        (*target_y)  << 0.1, 0.2, 0.3, 0.5, 0.6, 0.1;
        (*predicted_y) << 0.2, 0.2, 0.4, 0.4, 0.5, 0.6;
    }

private:
    // Target y values.
    mic::types::MatrixPtr<float> target_y;

    // Predicted y values.
    mic::types::MatrixPtr<float> predicted_y;
};


#endif /* LOSSTESTS_HPP_ */