Cheetah - SKA - PSS - Prototype Time Domain Search Pipeline
TimeFrequencyStatsTest.cpp
1 /*
2  * The MIT License (MIT)
3  *
4  * Copyright (c) 2016 The SKA organisation
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to deal
8  * in the Software without restriction, including without limitation the rights
9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10  * copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #include "cheetah/data/test/TimeFrequencyStatsTest.h"
25 #include "cheetah/generators/GaussianNoise.h"
26 
27 
28 namespace ska {
29 namespace cheetah {
30 namespace data {
31 namespace test {
32 
33 
34 TimeFrequencyStatsTest::TimeFrequencyStatsTest()
35  : ::testing::Test()
36 {
37 }
38 
39 TimeFrequencyStatsTest::~TimeFrequencyStatsTest()
40 {
41 }
42 
43 void TimeFrequencyStatsTest::SetUp()
44 {
45 }
46 
47 void TimeFrequencyStatsTest::TearDown()
48 {
49 }
50 
51 TEST_F(TimeFrequencyStatsTest, test_stat_calculation)
52 {
53  typedef uint8_t NumericalT;
54  generators::GaussianNoiseConfig generator_config;
55  generator_config.mean(5.0);
56  generator_config.std_deviation(1.0);
57  generators::GaussianNoise<NumericalT> generator(generator_config);
58 
59  std::size_t number_of_channels = 4100;
60  std::size_t number_of_spectra = 4096;
61 
62  TimeFrequency<Cpu, NumericalT> tf1((data::DimensionSize<data::Time>(number_of_spectra))
63  ,data::DimensionSize<data::Frequency>(number_of_channels));
64 
65  // fill data
66  generator.next(tf1);
67 
68  //Generate Stats block
69  TimeFrequencyStats<decltype(tf1)> stats(std::make_shared<TimeFrequency<Cpu, NumericalT>>(tf1));
70 
71  // Check statistics
72  auto const& channel_stats = stats.channel_stats();
73  ASSERT_EQ(channel_stats.size(), number_of_channels);
74  for(auto const& stat : channel_stats) {
75  ASSERT_NEAR(stat.mean, 5.0, 1.0);
76  ASSERT_NEAR(stat.variance, 1.0, 0.25);
77  }
78 
79  auto const& spectrum_stats = stats.spectrum_stats();
80  ASSERT_EQ(spectrum_stats.size(), number_of_spectra);
81  for(auto const& stat : spectrum_stats) {
82  ASSERT_NEAR(stat.mean, 5.0, 1.0);
83  ASSERT_NEAR(stat.variance, 1.0, 0.25);
84  }
85 
86 }
87 
88 } // namespace test
89 } // namespace data
90 } // namespace cheetah
91 } // namespace ska
Some limits and constants for FLDO.
Definition: Brdz.h:35