iomgr/time_averaged_stats_test.cc
Go to the documentation of this file.
1 /*
2  *
3  * Copyright 2015 gRPC authors.
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  */
18 
19 #include "src/core/lib/iomgr/time_averaged_stats.h"
20 
21 #include <math.h>
22 
23 #include <grpc/support/log.h>
24 
25 #include "test/core/util/test_config.h"
26 
27 #define EXPECT_EQ(a, b) GPR_ASSERT((a) == (b))
28 #define EXPECT_DOUBLE_EQ(a, b) GPR_ASSERT(fabs((a) - (b)) < 1e-9)
29 
30 static void no_regress_no_persist_test_1(void) {
31  grpc_time_averaged_stats tas;
32  grpc_time_averaged_stats_init(&tas, 1000, 0, 0.0);
33  EXPECT_DOUBLE_EQ(1000, tas.aggregate_weighted_avg);
34  EXPECT_DOUBLE_EQ(0, tas.aggregate_total_weight);
35 
36  /* Should have no effect */
37  grpc_time_averaged_stats_update_average(&tas);
38  EXPECT_DOUBLE_EQ(1000, tas.aggregate_weighted_avg);
39  EXPECT_DOUBLE_EQ(0, tas.aggregate_total_weight);
40 
41  /* Should replace old average */
42  grpc_time_averaged_stats_add_sample(&tas, 2000);
43  grpc_time_averaged_stats_update_average(&tas);
44  EXPECT_DOUBLE_EQ(2000, tas.aggregate_weighted_avg);
45  EXPECT_DOUBLE_EQ(1, tas.aggregate_total_weight);
46 }
47 
48 static void no_regress_no_persist_test_2(void) {
49  grpc_time_averaged_stats tas;
50  grpc_time_averaged_stats_init(&tas, 1000, 0, 0.0);
51  EXPECT_DOUBLE_EQ(1000, tas.aggregate_weighted_avg);
52  /* Should replace init value */
53  grpc_time_averaged_stats_add_sample(&tas, 2000);
54  grpc_time_averaged_stats_update_average(&tas);
55  EXPECT_DOUBLE_EQ(2000, tas.aggregate_weighted_avg);
56  EXPECT_DOUBLE_EQ(1, tas.aggregate_total_weight);
57 
58  grpc_time_averaged_stats_add_sample(&tas, 3000);
59  grpc_time_averaged_stats_update_average(&tas);
60  EXPECT_DOUBLE_EQ(3000, tas.aggregate_weighted_avg);
61  EXPECT_DOUBLE_EQ(1, tas.aggregate_total_weight);
62 }
63 
64 static void no_regress_no_persist_test_3(void) {
65  grpc_time_averaged_stats tas;
66  grpc_time_averaged_stats_init(&tas, 1000, 0, 0.0);
67  EXPECT_DOUBLE_EQ(1000, tas.aggregate_weighted_avg);
68  /* Should replace init value */
69  grpc_time_averaged_stats_add_sample(&tas, 2500);
70  grpc_time_averaged_stats_update_average(&tas);
71  EXPECT_DOUBLE_EQ(2500, tas.aggregate_weighted_avg);
72  EXPECT_DOUBLE_EQ(1, tas.aggregate_total_weight);
73 
74  grpc_time_averaged_stats_add_sample(&tas, 3500);
75  grpc_time_averaged_stats_add_sample(&tas, 4500);
76  grpc_time_averaged_stats_update_average(&tas);
77  EXPECT_DOUBLE_EQ(4000, tas.aggregate_weighted_avg);
78  EXPECT_DOUBLE_EQ(2, tas.aggregate_total_weight);
79 }
80 
81 static void some_regress_no_persist_test(void) {
82  grpc_time_averaged_stats tas;
83  grpc_time_averaged_stats_init(&tas, 1000, 0.5, 0.0);
84  EXPECT_DOUBLE_EQ(1000, tas.aggregate_weighted_avg);
85  EXPECT_DOUBLE_EQ(0, tas.aggregate_total_weight);
86  grpc_time_averaged_stats_add_sample(&tas, 2000);
87  grpc_time_averaged_stats_add_sample(&tas, 2000);
88  grpc_time_averaged_stats_update_average(&tas);
89  /* (2 * 2000 + 0.5 * 1000) / 2.5 */
90  EXPECT_DOUBLE_EQ(1800, tas.aggregate_weighted_avg);
91  EXPECT_DOUBLE_EQ(2.5, tas.aggregate_total_weight);
92 }
93 
94 static void some_decay_test(void) {
95  grpc_time_averaged_stats tas;
96  grpc_time_averaged_stats_init(&tas, 1000, 1, 0.0);
97  EXPECT_EQ(1000, tas.aggregate_weighted_avg);
98  /* Should avg with init value */
99  grpc_time_averaged_stats_add_sample(&tas, 2000);
100  grpc_time_averaged_stats_update_average(&tas);
101  EXPECT_DOUBLE_EQ(1500, tas.aggregate_weighted_avg);
102  EXPECT_DOUBLE_EQ(2, tas.aggregate_total_weight);
103 
104  grpc_time_averaged_stats_add_sample(&tas, 2000);
105  grpc_time_averaged_stats_update_average(&tas);
106  EXPECT_DOUBLE_EQ(1500, tas.aggregate_weighted_avg);
107  EXPECT_DOUBLE_EQ(2, tas.aggregate_total_weight);
108 
109  grpc_time_averaged_stats_add_sample(&tas, 2000);
110  grpc_time_averaged_stats_update_average(&tas);
111  EXPECT_DOUBLE_EQ(1500, tas.aggregate_weighted_avg);
112  EXPECT_DOUBLE_EQ(2, tas.aggregate_total_weight);
113 }
114 
115 static void no_regress_full_persist_test(void) {
116  grpc_time_averaged_stats tas;
117  grpc_time_averaged_stats_init(&tas, 1000, 0, 1.0);
118  EXPECT_DOUBLE_EQ(1000, tas.aggregate_weighted_avg);
119  EXPECT_DOUBLE_EQ(0, tas.aggregate_total_weight);
120 
121  /* Should replace init value */
122  grpc_time_averaged_stats_add_sample(&tas, 2000);
123  grpc_time_averaged_stats_update_average(&tas);
124  EXPECT_EQ(2000, tas.aggregate_weighted_avg);
125  EXPECT_EQ(1, tas.aggregate_total_weight);
126 
127  /* Will result in average of the 3 samples. */
128  grpc_time_averaged_stats_add_sample(&tas, 2300);
129  grpc_time_averaged_stats_add_sample(&tas, 2300);
130  grpc_time_averaged_stats_update_average(&tas);
131  EXPECT_DOUBLE_EQ(2200, tas.aggregate_weighted_avg);
132  EXPECT_DOUBLE_EQ(3, tas.aggregate_total_weight);
133 }
134 
135 static void no_regress_some_persist_test(void) {
136  grpc_time_averaged_stats tas;
137  grpc_time_averaged_stats_init(&tas, 1000, 0, 0.5);
138  /* Should replace init value */
139  grpc_time_averaged_stats_add_sample(&tas, 2000);
140  grpc_time_averaged_stats_update_average(&tas);
141  EXPECT_DOUBLE_EQ(2000, tas.aggregate_weighted_avg);
142  EXPECT_DOUBLE_EQ(1, tas.aggregate_total_weight);
143 
144  grpc_time_averaged_stats_add_sample(&tas, 2500);
145  grpc_time_averaged_stats_add_sample(&tas, 4000);
146  grpc_time_averaged_stats_update_average(&tas);
147  EXPECT_DOUBLE_EQ(3000, tas.aggregate_weighted_avg);
148  EXPECT_DOUBLE_EQ(2.5, tas.aggregate_total_weight);
149 }
150 
151 static void some_regress_some_persist_test(void) {
152  grpc_time_averaged_stats tas;
153  grpc_time_averaged_stats_init(&tas, 1000, 0.4, 0.6);
154  /* Sample weight = 0 */
155  EXPECT_EQ(1000, tas.aggregate_weighted_avg);
156  EXPECT_EQ(0, tas.aggregate_total_weight);
157 
158  grpc_time_averaged_stats_update_average(&tas);
159  /* (0.6 * 0 * 1000 + 0.4 * 1000 / 0.4) */
160  EXPECT_DOUBLE_EQ(1000, tas.aggregate_weighted_avg);
161  EXPECT_DOUBLE_EQ(0.4, tas.aggregate_total_weight);
162 
163  grpc_time_averaged_stats_add_sample(&tas, 2640);
164  grpc_time_averaged_stats_update_average(&tas);
165  /* (1 * 2640 + 0.6 * 0.4 * 1000 + 0.4 * 1000 / (1 + 0.6 * 0.4 + 0.4) */
166  EXPECT_DOUBLE_EQ(2000, tas.aggregate_weighted_avg);
167  EXPECT_DOUBLE_EQ(1.64, tas.aggregate_total_weight);
168 
169  grpc_time_averaged_stats_add_sample(&tas, 2876.8);
170  grpc_time_averaged_stats_update_average(&tas);
171  /* (1 * 2876.8 + 0.6 * 1.64 * 2000 + 0.4 * 1000 / (1 + 0.6 * 1.64 + 0.4) */
172  EXPECT_DOUBLE_EQ(2200, tas.aggregate_weighted_avg);
173  EXPECT_DOUBLE_EQ(2.384, tas.aggregate_total_weight);
174 
175  grpc_time_averaged_stats_add_sample(&tas, 4944.32);
176  grpc_time_averaged_stats_update_average(&tas);
177  /* (1 * 4944.32 + 0.6 * 2.384 * 2200 + 0.4 * 1000) /
178  (1 + 0.6 * 2.384 + 0.4) */
179  EXPECT_DOUBLE_EQ(3000, tas.aggregate_weighted_avg);
180  EXPECT_DOUBLE_EQ(2.8304, tas.aggregate_total_weight);
181 }
182 
183 int main(int argc, char** argv) {
184  grpc::testing::TestEnvironment env(&argc, argv);
185  no_regress_no_persist_test_1();
186  no_regress_no_persist_test_2();
187  no_regress_no_persist_test_3();
188  some_regress_no_persist_test();
189  some_decay_test();
190  no_regress_full_persist_test();
191  no_regress_some_persist_test();
192  some_regress_some_persist_test();
193  return 0;
194 }
some_regress_no_persist_test
static void some_regress_no_persist_test(void)
Definition: iomgr/time_averaged_stats_test.cc:81
log.h
generate.env
env
Definition: generate.py:37
grpc_time_averaged_stats_init
void grpc_time_averaged_stats_init(grpc_time_averaged_stats *stats, double init_avg, double regress_weight, double persistence_factor)
Definition: iomgr/time_averaged_stats.cc:23
no_regress_no_persist_test_1
static void no_regress_no_persist_test_1(void)
Definition: iomgr/time_averaged_stats_test.cc:30
no_regress_full_persist_test
static void no_regress_full_persist_test(void)
Definition: iomgr/time_averaged_stats_test.cc:115
grpc_time_averaged_stats
Definition: iomgr/time_averaged_stats.h:27
EXPECT_EQ
#define EXPECT_EQ(a, b)
Definition: iomgr/time_averaged_stats_test.cc:27
grpc_time_averaged_stats::aggregate_total_weight
double aggregate_total_weight
Definition: iomgr/time_averaged_stats.h:55
grpc_time_averaged_stats_add_sample
void grpc_time_averaged_stats_add_sample(grpc_time_averaged_stats *stats, double value)
Definition: iomgr/time_averaged_stats.cc:35
grpc_time_averaged_stats_update_average
double grpc_time_averaged_stats_update_average(grpc_time_averaged_stats *stats)
Definition: iomgr/time_averaged_stats.cc:41
main
int main(int argc, char **argv)
Definition: iomgr/time_averaged_stats_test.cc:183
no_regress_no_persist_test_2
static void no_regress_no_persist_test_2(void)
Definition: iomgr/time_averaged_stats_test.cc:48
some_decay_test
static void some_decay_test(void)
Definition: iomgr/time_averaged_stats_test.cc:94
no_regress_some_persist_test
static void no_regress_some_persist_test(void)
Definition: iomgr/time_averaged_stats_test.cc:135
test_config.h
time_averaged_stats.h
grpc::testing::TestEnvironment
Definition: test/core/util/test_config.h:54
EXPECT_DOUBLE_EQ
#define EXPECT_DOUBLE_EQ(a, b)
Definition: iomgr/time_averaged_stats_test.cc:28
some_regress_some_persist_test
static void some_regress_some_persist_test(void)
Definition: iomgr/time_averaged_stats_test.cc:151
no_regress_no_persist_test_3
static void no_regress_no_persist_test_3(void)
Definition: iomgr/time_averaged_stats_test.cc:64
grpc_time_averaged_stats::aggregate_weighted_avg
double aggregate_weighted_avg
Definition: iomgr/time_averaged_stats.h:58

grpc
Author(s):
autogenerated on Fri May 16 2025 03:00:36