From: Daniel Latypov <dlatypov@google.com>
To: andriy.shevchenko@linux.intel.com
Cc: brendanhiggins@google.com, davidgow@google.com,
linux-kernel@vger.kernel.org, linux-kselftest@vger.kernel.org,
skhan@linuxfoundation.org, Daniel Latypov <dlatypov@google.com>
Subject: [PATCH] lib: add basic KUnit test for lib/math
Date: Mon, 19 Oct 2020 15:45:56 -0700 [thread overview]
Message-ID: <20201019224556.3536790-1-dlatypov@google.com> (raw)
Add basic test coverage for files that don't require any config options:
* gcd.c
* lcm.c
* int_sqrt.c
* reciprocal_div.c
(Ignored int_pow.c since it's a simple textbook algorithm.)
These tests aren't particularly interesting, but
* they're chosen as easy to understand examples of how to write tests
* provides a place to add tests for any new files in this dir
* written so adding new test cases to cover edge cases should be easy
Signed-off-by: Daniel Latypov <dlatypov@google.com>
---
lib/math/Kconfig | 5 ++
lib/math/Makefile | 2 +
lib/math/math_test.c | 197 +++++++++++++++++++++++++++++++++++++++++++
3 files changed, 204 insertions(+)
create mode 100644 lib/math/math_test.c
diff --git a/lib/math/Kconfig b/lib/math/Kconfig
index f19bc9734fa7..6ba8680439c1 100644
--- a/lib/math/Kconfig
+++ b/lib/math/Kconfig
@@ -15,3 +15,8 @@ config PRIME_NUMBERS
config RATIONAL
bool
+
+config MATH_KUNIT_TEST
+ tristate "KUnit test for lib/math" if !KUNIT_ALL_TESTS
+ default KUNIT_ALL_TESTS
+ depends on KUNIT
diff --git a/lib/math/Makefile b/lib/math/Makefile
index be6909e943bd..fba6fe90f50b 100644
--- a/lib/math/Makefile
+++ b/lib/math/Makefile
@@ -4,3 +4,5 @@ obj-y += div64.o gcd.o lcm.o int_pow.o int_sqrt.o reciprocal_div.o
obj-$(CONFIG_CORDIC) += cordic.o
obj-$(CONFIG_PRIME_NUMBERS) += prime_numbers.o
obj-$(CONFIG_RATIONAL) += rational.o
+
+obj-$(CONFIG_MATH_KUNIT_TEST) += math_test.o
diff --git a/lib/math/math_test.c b/lib/math/math_test.c
new file mode 100644
index 000000000000..6f4681ea7c72
--- /dev/null
+++ b/lib/math/math_test.c
@@ -0,0 +1,197 @@
+// SPDX-License-Identifier: GPL-2.0
+/*
+ * Simple KUnit suite for math helper funcs that are always enabled.
+ *
+ * Copyright (C) 2020, Google LLC.
+ * Author: Daniel Latypov <dlatypov@google.com>
+ */
+
+#include <kunit/test.h>
+#include <linux/gcd.h>
+#include <linux/kernel.h>
+#include <linux/lcm.h>
+#include <linux/reciprocal_div.h>
+
+/* Generic test case for unsigned long inputs. */
+struct test_case {
+ unsigned long a, b;
+ unsigned long result;
+};
+
+static void gcd_test(struct kunit *test)
+{
+ const char *message_fmt = "gcd(%lu, %lu)";
+ int i;
+
+ struct test_case test_cases[] = {
+ {
+ .a = 0, .b = 1,
+ .result = 1,
+ },
+ {
+ .a = 2, .b = 2,
+ .result = 2,
+ },
+ {
+ .a = 2, .b = 4,
+ .result = 2,
+ },
+ {
+ .a = 3, .b = 5,
+ .result = 1,
+ },
+ {
+ .a = 3*9, .b = 3*5,
+ .result = 3,
+ },
+ {
+ .a = 3*5*7, .b = 3*5*11,
+ .result = 15,
+ },
+ {
+ .a = (1 << 21) - 1,
+ .b = (1 << 22) - 1,
+ .result = 1,
+ },
+ };
+
+ for (i = 0; i < ARRAY_SIZE(test_cases); ++i) {
+ KUNIT_EXPECT_EQ_MSG(test, test_cases[i].result,
+ gcd(test_cases[i].a, test_cases[i].b),
+ message_fmt, test_cases[i].a,
+ test_cases[i].b);
+
+ /* gcd(a,b) == gcd(b,a) */
+ KUNIT_EXPECT_EQ_MSG(test, test_cases[i].result,
+ gcd(test_cases[i].b, test_cases[i].a),
+ message_fmt, test_cases[i].b,
+ test_cases[i].a);
+ }
+}
+
+static void lcm_test(struct kunit *test)
+{
+ const char *message_fmt = "lcm(%lu, %lu)";
+ int i;
+
+ struct test_case test_cases[] = {
+ {
+ .a = 0, .b = 1,
+ .result = 0,
+ },
+ {
+ .a = 1, .b = 2,
+ .result = 2,
+ },
+ {
+ .a = 2, .b = 2,
+ .result = 2,
+ },
+ {
+ .a = 3*5, .b = 3*7,
+ .result = 3*5*7,
+ },
+ };
+
+ for (i = 0; i < ARRAY_SIZE(test_cases); ++i) {
+ KUNIT_EXPECT_EQ_MSG(test, test_cases[i].result,
+ lcm(test_cases[i].a, test_cases[i].b),
+ message_fmt, test_cases[i].a,
+ test_cases[i].b);
+
+ /* lcm(a,b) == lcm(b,a) */
+ KUNIT_EXPECT_EQ_MSG(test, test_cases[i].result,
+ lcm(test_cases[i].b, test_cases[i].a),
+ message_fmt, test_cases[i].b,
+ test_cases[i].a);
+ }
+}
+
+static void int_sqrt_test(struct kunit *test)
+{
+ const char *message_fmt = "sqrt(%lu)";
+ int i;
+
+ struct test_case test_cases[] = {
+ {
+ .a = 0,
+ .result = 0,
+ },
+ {
+ .a = 1,
+ .result = 1,
+ },
+ {
+ .a = 4,
+ .result = 2,
+ },
+ {
+ .a = 5,
+ .result = 2,
+ },
+ {
+ .a = 8,
+ .result = 2,
+ },
+ {
+ .a = 1UL >> 32,
+ .result = 1UL >> 16,
+ },
+ };
+
+ for (i = 0; i < ARRAY_SIZE(test_cases); ++i) {
+ KUNIT_EXPECT_EQ_MSG(test, int_sqrt(test_cases[i].a),
+ test_cases[i].result, message_fmt,
+ test_cases[i].a);
+ }
+}
+
+struct reciprocal_test_case {
+ u32 a, b;
+ u32 result;
+};
+
+static void reciprocal_div_test(struct kunit *test)
+{
+ int i;
+ struct reciprocal_value rv;
+ struct reciprocal_test_case test_cases[] = {
+ {
+ .a = 0, .b = 1,
+ .result = 0,
+ },
+ {
+ .a = 42, .b = 20,
+ .result = 2,
+ },
+ {
+ .a = (1<<16), .b = (1<<14),
+ .result = 1<<2,
+ },
+ };
+
+ for (i = 0; i < ARRAY_SIZE(test_cases); ++i) {
+ rv = reciprocal_value(test_cases[i].b);
+ KUNIT_EXPECT_EQ_MSG(test, test_cases[i].result,
+ reciprocal_divide(test_cases[i].a, rv),
+ "reciprocal_divide(%u, %u)",
+ test_cases[i].a, test_cases[i].b);
+ }
+}
+
+static struct kunit_case math_test_cases[] = {
+ KUNIT_CASE(gcd_test),
+ KUNIT_CASE(lcm_test),
+ KUNIT_CASE(int_sqrt_test),
+ KUNIT_CASE(reciprocal_div_test),
+ {}
+};
+
+static struct kunit_suite math_test_suite = {
+ .name = "lib-math",
+ .test_cases = math_test_cases,
+};
+
+kunit_test_suites(&math_test_suite);
+
+MODULE_LICENSE("GPL v2");
base-commit: 7cf726a59435301046250c42131554d9ccc566b8
--
2.29.0.rc1.297.gfa9743e501-goog
next reply other threads:[~2020-10-19 22:46 UTC|newest]
Thread overview: 17+ messages / expand[flat|nested] mbox.gz Atom feed top
2020-10-19 22:45 Daniel Latypov [this message]
2020-10-20 0:45 ` [PATCH] lib: add basic KUnit test for lib/math kernel test robot
2020-10-20 8:09 ` Andy Shevchenko
2020-10-20 16:13 ` Daniel Latypov
2020-10-21 3:40 ` David Gow
2020-10-21 17:47 ` Daniel Latypov
2020-10-22 15:06 ` Andy Shevchenko
2020-10-22 16:26 ` Daniel Latypov
2020-10-22 18:51 ` Brendan Higgins
2020-10-22 19:10 ` Andy Shevchenko
2020-10-22 19:12 ` Andy Shevchenko
2020-10-22 18:53 ` Brendan Higgins
2020-10-22 19:05 ` Andy Shevchenko
2020-10-22 21:21 ` Brendan Higgins
2020-10-23 9:02 ` Andy Shevchenko
2020-11-02 14:51 ` kernel test robot
2020-11-03 1:32 ` kernel test robot
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=20201019224556.3536790-1-dlatypov@google.com \
--to=dlatypov@google.com \
--cc=andriy.shevchenko@linux.intel.com \
--cc=brendanhiggins@google.com \
--cc=davidgow@google.com \
--cc=linux-kernel@vger.kernel.org \
--cc=linux-kselftest@vger.kernel.org \
--cc=skhan@linuxfoundation.org \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for NNTP newsgroup(s).