linux/fs/bcachefs/mean_and_variance_test.c

// SPDX-License-Identifier: GPL-2.0
#include <kunit/test.h>

#include "mean_and_variance.h"

#define MAX_SQR …

static void mean_and_variance_basic_test(struct kunit *test)
{ … }

/*
 * Test values computed using a spreadsheet from the psuedocode at the bottom:
 * https://fanf2.user.srcf.net/hermes/doc/antiforgery/stats.pdf
 */

static void mean_and_variance_weighted_test(struct kunit *test)
{ … }

static void mean_and_variance_weighted_advanced_test(struct kunit *test)
{ … }

static void do_mean_and_variance_test(struct kunit *test,
				      s64 initial_value,
				      s64 initial_n,
				      s64 n,
				      unsigned weight,
				      s64 *data,
				      s64 *mean,
				      s64 *stddev,
				      s64 *weighted_mean,
				      s64 *weighted_stddev)
{ … }

/* Test behaviour with a single outlier, then back to steady state: */
static void mean_and_variance_test_1(struct kunit *test)
{ … }

/* Test behaviour where we switch from one steady state to another: */
static void mean_and_variance_test_2(struct kunit *test)
{ … }

static void mean_and_variance_fast_divpow2(struct kunit *test)
{ … }

static void mean_and_variance_u128_basic_test(struct kunit *test)
{ … }

static struct kunit_case mean_and_variance_test_cases[] = …;

static struct kunit_suite mean_and_variance_test_suite = …;

kunit_test_suite(…) …;

MODULE_AUTHOR(…) …;
MODULE_DESCRIPTION(…) …;
MODULE_LICENSE(…) …;