@tensorflow/tfjs-node/dist/run_tests.js

"use strict";
/**
 * @license
 * Copyright 2018 Google LLC. All Rights Reserved.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * =============================================================================
 */
Object.defineProperty(exports, "__esModule", { value: true });
// We import index.ts so that the Node backend gets registered.
require("./index");
var tf = require("@tensorflow/tfjs");
// tslint:disable-next-line:no-imports-from-dist
var jasmine_util = require("@tensorflow/tfjs-core/dist/jasmine_util");
var yargs_1 = require("yargs");
Error.stackTraceLimit = Infinity;
// tslint:disable-next-line:no-require-imports
var jasmineCtor = require('jasmine');
// tslint:disable-next-line:no-require-imports
process.on('unhandledRejection', function (e) {
    throw e;
});
jasmine_util.setTestEnvs([{
        name: 'test-tensorflow',
        backendName: 'tensorflow',
        flags: {},
        isDataSync: true
    }]);
var IGNORE_LIST = [
    // Always ignore version tests:
    'version version',
    'unreliable is true due to both auto gc and string tensors',
    'unreliable is true due to auto gc',
    // See https://github.com/tensorflow/tfjs/issues/161
    'depthwiseConv2D',
    'separableConv2d',
    'complex64 memory',
    // See https://github.com/tensorflow/tfjs-core/pull/1270
    'depthToSpace test-tensorflow {} throws when blocksize < 2',
    // tslint:disable-next-line:max-line-length
    'depthToSpace test-tensorflow {} throws when CPU backend used with data format NCHW',
    // See https://github.com/tensorflow/tfjs/issues/806
    'scatterND test-tensorflow {} should work for 2d',
    'scatterND test-tensorflow {} should work for simple 1d',
    'scatterND test-tensorflow {} should work for multiple 1d',
    'scatterND test-tensorflow {} should sum the duplicated indices',
    'scatterND test-tensorflow {} should work for tensorLike input',
    // https://github.com/tensorflow/tfjs/issues/1077
    'maxPool test-tensorflow {} x=[2,2,3] f=[1,1] s=2 p=1 dimRoundingMode=floor',
    'avgPool test-tensorflow {} x=[2,2,3] f=[1,1] s=2 p=1 dimRoundingMode=floor',
    // tslint:disable-next-line:max-line-length
    'avgPool3d test-tensorflow {} x=[1,2,2,2,1] f=[2,2,2] s=1 p=1 roundingMode=floor',
    // tslint:disable-next-line:max-line-length
    'maxPool3d test-tensorflow {} x=[1,2,2,2,1] f=[2,2,2] s=1 p=1 roundingMode=floor',
    // libtensorflow doesn't support 6D ArgMax yet.
    'argmax test-tensorflow {} 6D, axis=0', 'diag test-tensorflow {} complex',
    'diag test-tensorflow {} bool',
    // See https://github.com/tensorflow/tfjs/issues/1891
    'conv2d test-tensorflow {} x=[2,1,2,2] f=[1,1,1,1] s=1 d=1 p=0 NCHW',
    'conv2d test-tensorflow {} x=[1,2,2] f=[2,2,1,1] s=1 d=1 p=explicit NCHW',
    'conv2d test-tensorflow {} x=[1,2,2] f=[2,2,1,1] s=1 d=1 p=same NCHW',
    'conv2d test-tensorflow {} x=[2,2,2] f=[2,2,2,1] s=1 d=1 p=same NCHW',
    'conv2d test-tensorflow {} x=[2,1,2,2] f=[2,2,1,1] s=1 d=1 p=same NCHW',
    'conv2d test-tensorflow {} gradient x=[1,1,3,3] f=[2,2,1,1] s=1 p=0 NCHW',
    'conv2d test-tensorflow {} gradient x=[2,1,3,3] f=[2,2,1,1] s=1 p=0 NCHW',
    'maxPoolWithArgmax', 'rotate', 'flipLeftRight'
];
if (process.platform === 'win32') {
    // Ignore tensorboard on windows because tmp folder cleanup used in tests
    // fails with tmp folder cleanup:
    // https://github.com/tensorflow/tfjs/issues/1692
    IGNORE_LIST.push('tensorboard');
    IGNORE_LIST.push('tensorBoard');
    // Windows has two failing tests:
    // https://github.com/tensorflow/tfjs/issues/598
    IGNORE_LIST.push('clip test-tensorflow {} propagates NaNs');
    IGNORE_LIST.push('maxPool test-tensorflow {} [x=[3,3,1] f=[2,2] s=1 ignores NaNs');
}
var runner = new jasmineCtor();
runner.loadConfig({ spec_files: ['src/**/*_test.ts'], random: false });
// Also import tests from core.
// tslint:disable-next-line: no-imports-from-dist
require("@tensorflow/tfjs-core/dist/tests");
if (process.env.JASMINE_SEED) {
    runner.seed(process.env.JASMINE_SEED);
}
var env = jasmine.getEnv();
var grepRegex = new RegExp(yargs_1.argv.grep);
// Filter method that returns boolean, if a given test should return.
env.specFilter = function (spec) {
    // Filter based on the grep flag.
    if (!grepRegex.test(spec.getFullName())) {
        return false;
    }
    // Return false (skip the test) if the test is in the ignore list.
    for (var i = 0; i < IGNORE_LIST.length; ++i) {
        if (spec.getFullName().indexOf(IGNORE_LIST[i]) > -1) {
            return false;
        }
    }
    // Otherwise run the test.
    return true;
};
// TODO(kreeger): Consider moving to C-code.
console.log("Running tests against TensorFlow: " + tf.backend().binding.TF_Version);
runner.execute();