Function setUpFakes

speech-commands/src/browser_fft_recognizer_test.ts:61–95

(model?: tfl.Sequential, backgroundAndNoiseOnly = false)

  let tfLoadModelSpy: jasmine.Spy;

  function setUpFakes(model?: tfl.Sequential, backgroundAndNoiseOnly = false) {
    const words =
        backgroundAndNoiseOnly ? fakeWordsNoiseAndUnknownOnly : fakeWords;
    const numWords = words.length;
    tfLoadModelSpy =
        spyOn(tfl, 'loadLayersModel').and.callFake((url: string) => {
          if (model == null) {
            model = tfl.sequential();
            model.add(tfl.layers.flatten(
                {inputShape: [fakeNumFrames, fakeColumnTruncateLength, 1]}));
            secondLastBaseDenseLayer = tfl.layers.dense({
              units: 4,
              activation: 'relu',
              kernelInitializer: 'leCunNormal'
            });
            model.add(secondLastBaseDenseLayer);
            model.add(tfl.layers.dense({
              units: numWords,
              useBias: false,
              kernelInitializer: 'leCunNormal',
              activation: 'softmax'
            }));
          }
          return model;
        });
    spyOn(BrowserFftUtils, 'loadMetadataJson')
        .and.callFake(async (url: string) => {
          return {words};
        });

    spyOn(BrowserFftUtils, 'getAudioContextConstructor')
        .and.callFake(() => FakeAudioContext.createInstance);
    spyOn(BrowserFftUtils, 'getAudioMediaStream')
        .and.callFake(() => new FakeAudioMediaStream());
  }

  it('Constructor works', () => {
    const recognizer = new BrowserFftSpeechCommandRecognizer();