Chromium Code Reviews| Index: webrtc/tools/frame_analyzer/reference_less_video_analysis_unittest.cc |
| diff --git a/webrtc/tools/frame_analyzer/reference_less_video_analysis_unittest.cc b/webrtc/tools/frame_analyzer/reference_less_video_analysis_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..e6c1e1eaa2845069d7ff28600566b642bf54c95e |
| --- /dev/null |
| +++ b/webrtc/tools/frame_analyzer/reference_less_video_analysis_unittest.cc |
| @@ -0,0 +1,59 @@ |
| +/* |
| + * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. |
| + * |
| + * Use of this source code is governed by a BSD-style license |
| + * that can be found in the LICENSE file in the root of the source |
| + * tree. An additional intellectual property rights grant can be found |
| + * in the file PATENTS. All contributing project authors may |
| + * be found in the AUTHORS file in the root of the source tree. |
| + */ |
| +#include <string.h> |
| +#include <stdio.h> |
| +#include <cstring> |
| +#include <iostream> |
| +#include <vector> |
| + |
| +#include "webrtc/test/gtest.h" |
| +#include "webrtc/test/testsupport/fileutils.h" |
| +#include "webrtc/tools/frame_analyzer/reference_less_video_analysis_lib.h" |
| + |
| +std::vector<double> psnr_per_frame; |
| +std::vector<double> ssim_per_frame; |
| +std::string video_file = |
| + webrtc::test::ResourcePath("reference_less_video_test_file", "y4m"); |
|
kjellander_webrtc
2016/11/28 15:02:08
Please add tests for an invalid test file as well,
|
| +char video_file_name[200]; |
| + |
| +TEST(ReferenceLessVideoAnalysisTest, MatchComputedMetrics) { |
| + strcpy(video_file_name, video_file.c_str()); |
| + compute_metrics(video_file_name, &psnr_per_frame, &ssim_per_frame); |
| + |
| + EXPECT_EQ(74, (int)psnr_per_frame.size()); |
| + |
| + EXPECT_FLOAT_EQ(27.2268, psnr_per_frame[1]); |
| + EXPECT_FLOAT_EQ(24.980579, psnr_per_frame[5]); |
| + |
| + EXPECT_FLOAT_EQ(0.983759, ssim_per_frame[1]); |
| + EXPECT_FLOAT_EQ(0.979621, ssim_per_frame[5]); |
| +} |
| + |
| +TEST(ReferenceLessVideoAnalysisTest, MatchHeightWidthFps) { |
| + int height = 0, width = 0, fps = 0; |
| + strcpy(video_file_name, video_file.c_str()); |
| + get_height_width_fps(&height, &width, &fps, video_file_name); |
| + EXPECT_EQ(height, 720); |
| + EXPECT_EQ(width, 1280); |
| + EXPECT_EQ(fps, 25); |
| +} |
| + |
| +TEST(ReferenceLessVideoAnalysisTest, MatchIdenticalFrameClusters) { |
| + std::vector<int> identical_frame_clusters = |
| + find_frame_clusters(psnr_per_frame, ssim_per_frame); |
| + EXPECT_EQ(5, (int)identical_frame_clusters.size()); |
| + EXPECT_EQ(1, identical_frame_clusters[0]); |
| + EXPECT_EQ(1, identical_frame_clusters[4]); |
| +} |
| + |
| + |
|
kjellander_webrtc
2016/11/28 15:02:08
Remove blank lines.
|
| + |
| + |
| + |