Function TEST

tensorflow/core/kernels/decode_wav_op_test.cc:38–85

namespace {

TEST(DecodeWavOpTest, DecodeWavTest) {
  Scope root = Scope::NewRootScope();

  std::vector<uint8> wav_data = {
      'R',  'I',  'F', 'F', 44,  0,   0,   0,  // size of whole file - 8
      'W',  'A',  'V', 'E', 'f', 'm', 't', ' ', 16, 0, 0,
      0,                   // size of fmt block - 8: 24 - 8
      1,    0,             // format: PCM (1)
      1,    0,             // channels: 1
      0x13, 0x37, 0,   0,  // sample rate: 14099
      0x26, 0x6e, 0,   0,  // byte rate: 2 * 14099
      2,    0,             // block align: NumChannels * BytesPerSample
      16,   0,             // bits per sample: 2 * 8
      'd',  'a',  't', 'a', 8,   0,   0,   0,  // size of payload: 8
      0,    0,                                 // first sample: 0
      0xff, 0x3f,                              // second sample: 16383
      0xff, 0x7f,  // third sample: 32767 (saturated)
      0x00, 0x80,  // fourth sample: -32768 (saturated)
  };
  Tensor content_tensor =
      test::AsScalar<tstring>(string(wav_data.begin(), wav_data.end()));
  Output content_op =
      Const(root.WithOpName("content_op"), Input::Initializer(content_tensor));

  DecodeWav decode_wav_op =
      DecodeWav(root.WithOpName("decode_wav_op"), content_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(1, 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_EQ(14099, sample_rate);
}

TEST(DecodeWavOpTest, DecodeWav_ShapeFn) {
  ShapeInferenceTestOp op("DecodeWav");