reshape_test.hhs

/**
 * @author Jianan Lin (林家南)
 * Function of test unit for reshape
 */

function reshape_test() {

    *import math: reshape

    let a = [1, 2, 3, 4, 5, 6, 7, 8];
    let b = reshape(a, [2, 2, 2]).val;
    for (let i = 0; i < 2; i++) {
        for (let j = 0; j < 2; j++) {
            for (let k = 0; k < 2; k++) {
                if (b[i][j][k] !== 4 * i + 2 * j + k + 1) {
                    throw new Error('Failed unit test for reshape.');
                }
            }
        }
    }

    let c = reshape(a, [2, -1]).val;
    for (let i = 0; i < 2; i++) {
        for (let j = 0; j < 4; j++) {
            if (c[i][j] !== 4 * i + j + 1) {
                throw new Error('Failed unit test for reshape.');
            }
        }
    }

    //print('reshape test pass');
}