Function setupFakeWeightFiles

tfjs-node/src/io/node_http_test.ts:66–88

(fileBufferMap: {
    [filename: string]: string|Float32Array|Int32Array|ArrayBuffer|Uint8Array|
    Uint16Array
  })

describe('nodeHTTPRequest-load', () => {
  let requestInits: RequestInit[];
  const setupFakeWeightFiles = (fileBufferMap: {
    [filename: string]: string|Float32Array|Int32Array|ArrayBuffer|Uint8Array|
    Uint16Array
  }) => {
    spyOn(tf.env().platform, 'fetch')
        .and.callFake((path: string, init: RequestInit) => {
          return new Promise((resolve, reject) => {
            let contentType = '';
            if (path.endsWith('model.json')) {
              contentType = JSON_TYPE;
            } else if (
                path.endsWith('weightfile0') || path.endsWith('weightfile1')) {
              contentType = OCTET_STREAM_TYPE;
            } else {
              reject(new Error(`Invalid path: ${path}`));
            }
            requestInits.push(init);
            resolve(new fetch.Response(
                fileBufferMap[path],
                {'headers': {'Content-Type': contentType}}));
          });
        });
  };

  beforeEach(() => {
    requestInits = [];