Function TEST

tensorflow/core/kernels/mfcc_dct_test.cc:25–42

namespace tensorflow {

TEST(MfccDctTest, AgreesWithMatlab) {
  // This test verifies the DCT against MATLAB's dct function.
  MfccDct dct;
  std::vector<double> input = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0};
  const int kCoefficientCount = 6;
  ASSERT_TRUE(dct.Initialize(input.size(), kCoefficientCount));
  std::vector<double> output;
  dct.Compute(input, &output);
  // Note, the matlab dct function divides the first coefficient by
  // sqrt(2), whereas we don't, so we multiply the first element of
  // the matlab result by sqrt(2) to get the expected values below.
  std::vector<double> expected = {12.1243556530, -4.1625617959, 0.0,
                                  -0.4082482905, 0.0,           -0.0800788912};
  ASSERT_EQ(output.size(), kCoefficientCount);
  for (int i = 0; i < kCoefficientCount; ++i) {
    EXPECT_NEAR(output[i], expected[i], 1e-10);
  }
}

TEST(MfccDctTest, InitializeFailsOnInvalidInput) {
  MfccDct dct1;