1 // SPDX-License-Identifier: GPL-2.0
23 	KUNIT_EXPECT_EQ(test, mean_and_variance_get_variance(s), 1);  in mean_and_variance_basic_test()
28  * Test values computed using a spreadsheet from the psuedocode at the bottom:
50 	mean_and_variance_weighted_update(&s, -10);  in mean_and_variance_weighted_test()
51 	KUNIT_EXPECT_EQ(test, mean_and_variance_weighted_get_mean(s), -10);  in mean_and_variance_weighted_test()
54 	mean_and_variance_weighted_update(&s, -20);  in mean_and_variance_weighted_test()
55 	KUNIT_EXPECT_EQ(test, mean_and_variance_weighted_get_mean(s), -12);  in mean_and_variance_weighted_test()
58 	mean_and_variance_weighted_update(&s, -30);  in mean_and_variance_weighted_test()
59 	KUNIT_EXPECT_EQ(test, mean_and_variance_weighted_get_mean(s), -16);  in mean_and_variance_weighted_test()
76 	for (i = -10; i >= -100; i -= 10)  in mean_and_variance_weighted_advanced_test()
79 	KUNIT_EXPECT_EQ(test, mean_and_variance_weighted_get_mean(s), -11);  in mean_and_variance_weighted_advanced_test()
120 /* Test behaviour with a single outlier, then back to steady state: */
123 	s64 d[]			= { 100, 10, 10, 10, 10, 10, 10 };  in mean_and_variance_test_1()  local
129 	do_mean_and_variance_test(test, 10, 6, ARRAY_SIZE(d), 2,  in mean_and_variance_test_1()
130 			d, mean, stddev, weighted_mean, weighted_stddev);  in mean_and_variance_test_1()
135 	s64 d[]			= { 100, 10, 10, 10, 10, 10, 10 };  in mean_and_variance_test_2()  local
137 	s64 stddev[]		= {   9,  9,  9,  9,  9,  9,  9 };  in mean_and_variance_test_2()
141 	do_mean_and_variance_test(test, 10, 6, ARRAY_SIZE(d), 2,  in mean_and_variance_test_2()
142 			d, mean, stddev, weighted_mean, weighted_stddev);  in mean_and_variance_test_2()
148 	s64 d[]			= { 100, 100, 100, 100, 100 };  in mean_and_variance_test_3()  local
154 	do_mean_and_variance_test(test, 10, 6, ARRAY_SIZE(d), 2,  in mean_and_variance_test_3()
155 			d, mean, stddev, weighted_mean, weighted_stddev);  in mean_and_variance_test_3()
160 	s64 d[]			= { 100, 100, 100, 100, 100 };  in mean_and_variance_test_4()  local
162 	s64 stddev[]		= {   9,  13,  15,  17,  19 };  in mean_and_variance_test_4()
166 	do_mean_and_variance_test(test, 10, 6, ARRAY_SIZE(d), 2,  in mean_and_variance_test_4()
167 			d, mean, stddev, weighted_mean, weighted_stddev);  in mean_and_variance_test_4()
173 	u8 d;  in mean_and_variance_fast_divpow2()  local
176 		d = 0;  in mean_and_variance_fast_divpow2()
177 		KUNIT_EXPECT_EQ(test, fast_divpow2(i, d), div_u64(i, 1LLU << d));  in mean_and_variance_fast_divpow2()
178 		KUNIT_EXPECT_EQ(test, abs(fast_divpow2(-i, d)), div_u64(i, 1LLU << d));  in mean_and_variance_fast_divpow2()
179 		for (d = 1; d < 32; d++) {  in mean_and_variance_fast_divpow2()
180 			KUNIT_EXPECT_EQ_MSG(test, abs(fast_divpow2(i, d)),  in mean_and_variance_fast_divpow2()
181 					    div_u64(i, 1 << d), "%lld %u", i, d);  in mean_and_variance_fast_divpow2()
182 			KUNIT_EXPECT_EQ_MSG(test, abs(fast_divpow2(-i, d)),  in mean_and_variance_fast_divpow2()
183 					    div_u64(i, 1 << d), "%lld %u", -i, d);  in mean_and_variance_fast_divpow2()
190 	u128_u a  = u64s_to_u128(0, U64_MAX);  in mean_and_variance_u128_basic_test()  local
191 	u128_u a1 = u64s_to_u128(0, 1);  in mean_and_variance_u128_basic_test()
192 	u128_u b  = u64s_to_u128(1, 0);  in mean_and_variance_u128_basic_test()
193 	u128_u c  = u64s_to_u128(0, 1LLU << 63);  in mean_and_variance_u128_basic_test()
196 	KUNIT_EXPECT_EQ(test, u128_hi(u128_add(a, a1)), 1);  in mean_and_variance_u128_basic_test()
197 	KUNIT_EXPECT_EQ(test, u128_lo(u128_add(a, a1)), 0);  in mean_and_variance_u128_basic_test()
198 	KUNIT_EXPECT_EQ(test, u128_hi(u128_add(a1, a)), 1);  in mean_and_variance_u128_basic_test()
199 	KUNIT_EXPECT_EQ(test, u128_lo(u128_add(a1, a)), 0);  in mean_and_variance_u128_basic_test()
204 	KUNIT_EXPECT_EQ(test, u128_hi(u128_shl(c, 1)), 1);  in mean_and_variance_u128_basic_test()
205 	KUNIT_EXPECT_EQ(test, u128_lo(u128_shl(c, 1)), 0);  in mean_and_variance_u128_basic_test()
207 	KUNIT_EXPECT_EQ(test, u128_hi(u128_square(U64_MAX)), U64_MAX - 1);  in mean_and_variance_u128_basic_test()
208 	KUNIT_EXPECT_EQ(test, u128_lo(u128_square(U64_MAX)), 1);  in mean_and_variance_u128_basic_test()
210 	KUNIT_EXPECT_EQ(test, u128_lo(u128_div(b, 2)), 1LLU << 63);  in mean_and_variance_u128_basic_test()
212 	KUNIT_EXPECT_EQ(test, u128_hi(u128_div(c2, 2)), U64_MAX >> 1);  in mean_and_variance_u128_basic_test()
215 	KUNIT_EXPECT_EQ(test, u128_hi(u128_div(u128_shl(u64_to_u128(U64_MAX), 32), 2)), U32_MAX >> 1);  in mean_and_variance_u128_basic_test()