test-bundle-size.js

#!/usr/bin/env node
// Copyright 2019 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.
// =============================================================================

// This script has not been bazelized yet because it uses several imports from
// the root `scripts/` directory.
const shell = require('shelljs');
const {exec} = require('../../scripts/test-util');
const {showDiff, getFileSizeBytes} = require('../../scripts/bundle-size-util');
const path = require('path');

// Get the bundle sizes from this change.
exec(`yarn build-npm`, {silent: false});

const basePath = '../dist/bin/tfjs-backend-wasm/tfjs-backend-wasm_pkg/dist/';
const bundleFilename = path.join(basePath, 'tf-backend-wasm.min.js');
const minBundleSize = getFileSizeBytes(bundleFilename);
const wasmFileName = path.join(basePath, 'tfjs-backend-wasm.wasm');
const wasmSize = getFileSizeBytes(wasmFileName);

console.log(`~~~~ WASM file ~~~~`);
console.log(`==> post-gzip`)
console.log(`size: ${wasmSize.gzipFileSizeBytes}`);
console.log();
console.log(`==> pre-gzip`)
console.log(`size: ${wasmSize.fileSizeBytes}`);
console.log();
console.log();

console.log(`~~~~ minified bundle (JavaScript) ~~~~`);
console.log(`==> post-gzip`)
console.log(`size: ${minBundleSize.gzipFileSizeBytes}`);
console.log();
console.log(`==> pre-gzip`)
console.log(`size: ${minBundleSize.fileSizeBytes}`);
console.log();
console.log();