ArrayTests.hpp

#ifndef UTENSOR_ARRAY_TESTS
#define UTENSOR_ARRAY_TESTS

#include "ArrayOps.hpp"
#include "test.hpp"
#include "tensorIdxImporter.hpp"

class ArrayOpsTest : public Test {
public:
    void quantize_v2Test(void) {
        testStart("quantize_v2");
        TensorIdxImporter t_import;
        
        //reference inputs  /Users/neitan01/Documents/mbed/uTensor.git/TESTS/scripts/PRE-GEN/qA
        Tensor<float> b = t_import.float_import ("/fs/testData/qB/in/Cast_1_0.idx");
        Tensor<float> b_min = t_import.float_import("/fs/testData/qB/in/Min_1_0.idx");
        Tensor<float> b_max = t_import.float_import("/fs/testData/qB/in/Max_1_0.idx");

        //reference outputs
        Tensor<unsigned char> b_q_ref = t_import.ubyte_import("/fs/testData/qB/out/qB_0.idx");
        Tensor<float> b_min_q_ref = t_import.float_import("/fs/testData/qB/out/qB_1.idx");
        Tensor<float> b_max_q_ref = t_import.float_import("/fs/testData/qB/out/qb_2.idx");

        Tensor<unsigned char> b_q(b_q_ref.getShape());
        Tensor<float> b_min_q(b_min_q_ref.getShape());
        Tensor<float> b_max_q(b_max_q_ref.getShape());

        //Implementation goes here
        timer_start();
        QuantizeV2(b, b_min, b_max, b_q, b_min_q, b_max_q);
        timer_stop();

        // printf("refMin is : %f \r\n", *(b_min_q_ref.getPointer({0})));
        // printf("outMin is : %f \r\n", *(b_min_q.getPointer({0})));
        // printf("diff : output(%f), outMin(%f), outMax(%f)\r\n", 
        //  meanPercentErr(b_q_ref, b_q), meanPercentErr(b_min_q_ref, b_min_q), meanPercentErr(b_max_q_ref, b_max_q));

        double result = meanPercentErr(b_q_ref, b_q) + meanPercentErr(b_min_q_ref, b_min_q) + meanPercentErr(b_max_q_ref, b_max_q);
        //passed(result < 0.0001);
        passed(result == 0);
    }

    void dequantizeTest(void) {
        testStart("dequantize");
        TensorIdxImporter t_import;

        //reference inputs
        Tensor<unsigned char> a = t_import.ubyte_import("/fs/testData/deQ/in/rQ_0.idx");
        Tensor<float> a_min = t_import.float_import("/fs/testData/deQ/in/rQ_1.idx");
        Tensor<float> a_max = t_import.float_import("/fs/testData/deQ/in/rQ_2.idx");

        //reference outputs
        Tensor<float> out_ref = t_import.float_import("/fs/testData/deQ/out/deQ_0.idx");

        //modify the checks below:
        Tensor<float> out(out_ref.getShape());

        timer_start();
        dequantize(a, a_min, a_max, out);
        timer_stop();

        double result = meanPercentErr(out_ref, out);
        //passed(result < 0.0001);
        passed(result == 0);
    }

    void reshapeTest(void) {
        testStart("reshape");
        TensorIdxImporter t_import;

        //reference inputs
        Tensor<float> ref_a = t_import.float_import("/fs/testData/ref_reshape/in/Const_0.idx");
        Tensor<int> ref_dim = t_import.int_import("/fs/testData/ref_reshape/in/Const_1_0.idx");

        //reference outputs
        Tensor<float> out_ref = t_import.float_import("/fs/testData/ref_reshape/out/ref_reshape_0.idx");

        //modify the checks below:
        Tensor<float> out(out_ref.getShape());

        timer_start();
        reshape(ref_a, ref_dim, out);
        timer_stop();

        double result = meanPercentErr(out_ref, out);
        //passed(result < 0.0001);
        passed(result == 0);
    }

    void runAll(void) {
        quantize_v2Test();
        dequantizeTest();
        reshapeTest();
    }
};


#endif //UTENSOR_ARRAY_TESTS