Function TEST

tensorflow/core/kernels/encode_wav_op_test.cc:37–78

namespace {

TEST(EncodeWavOpTest, EncodeWavTest) {
  Scope root = Scope::DisabledShapeInferenceScope();

  Tensor audio_tensor(DT_FLOAT, {4, 2});
  test::FillValues<float>(
      &audio_tensor, {0.0f, 0.5f, 1.0f, -1.0f, 0.25f, 0.75f, 1.25f, -0.5f});
  Output audio_op =
      Const(root.WithOpName("audio_op"), Input::Initializer(audio_tensor));

  Output sample_rate_op = Const(root.WithOpName("sample_rate_op"), 44100);

  EncodeWav encode_wav_op =
      EncodeWav(root.WithOpName("encode_wav_op"), audio_op, sample_rate_op);

  DecodeWav decode_wav_op =
      DecodeWav(root.WithOpName("decode_wav_op"), encode_wav_op);

  TF_ASSERT_OK(root.status());

  ClientSession session(root);
  std::vector<Tensor> outputs;

  TF_EXPECT_OK(session.Run(ClientSession::FeedType(),
                           {decode_wav_op.audio, decode_wav_op.sample_rate},
                           &outputs));

  const Tensor& audio = outputs[0];
  const int sample_rate = outputs[1].flat<int32>()(0);

  EXPECT_EQ(2, audio.dims());
  EXPECT_EQ(2, audio.dim_size(1));
  EXPECT_EQ(4, audio.dim_size(0));
  EXPECT_NEAR(0.0f, audio.flat<float>()(0), 1e-4f);
  EXPECT_NEAR(0.5f, audio.flat<float>()(1), 1e-4f);
  EXPECT_NEAR(1.0f, audio.flat<float>()(2), 1e-4f);
  EXPECT_NEAR(-1.0f, audio.flat<float>()(3), 1e-4f);
  EXPECT_NEAR(0.25f, audio.flat<float>()(4), 1e-4f);
  EXPECT_NEAR(0.75f, audio.flat<float>()(5), 1e-4f);
  EXPECT_NEAR(1.0f, audio.flat<float>()(6), 1e-4f);
  EXPECT_NEAR(-0.5f, audio.flat<float>()(7), 1e-4f);
  EXPECT_EQ(44100, sample_rate);
}

}  // namespace
}  // namespace ops