xref: /netbsd-src/external/bsd/jemalloc.old/dist/test/unit/bit_util.c (revision 8e33eff89e26cf71871ead62f0d5063e1313c33a)
1*8e33eff8Schristos #include "test/jemalloc_test.h"
2*8e33eff8Schristos 
3*8e33eff8Schristos #include "jemalloc/internal/bit_util.h"
4*8e33eff8Schristos 
5*8e33eff8Schristos #define TEST_POW2_CEIL(t, suf, pri) do {				\
6*8e33eff8Schristos 	unsigned i, pow2;						\
7*8e33eff8Schristos 	t x;								\
8*8e33eff8Schristos 									\
9*8e33eff8Schristos 	assert_##suf##_eq(pow2_ceil_##suf(0), 0, "Unexpected result");	\
10*8e33eff8Schristos 									\
11*8e33eff8Schristos 	for (i = 0; i < sizeof(t) * 8; i++) {				\
12*8e33eff8Schristos 		assert_##suf##_eq(pow2_ceil_##suf(((t)1) << i), ((t)1)	\
13*8e33eff8Schristos 		    << i, "Unexpected result");				\
14*8e33eff8Schristos 	}								\
15*8e33eff8Schristos 									\
16*8e33eff8Schristos 	for (i = 2; i < sizeof(t) * 8; i++) {				\
17*8e33eff8Schristos 		assert_##suf##_eq(pow2_ceil_##suf((((t)1) << i) - 1),	\
18*8e33eff8Schristos 		    ((t)1) << i, "Unexpected result");			\
19*8e33eff8Schristos 	}								\
20*8e33eff8Schristos 									\
21*8e33eff8Schristos 	for (i = 0; i < sizeof(t) * 8 - 1; i++) {			\
22*8e33eff8Schristos 		assert_##suf##_eq(pow2_ceil_##suf((((t)1) << i) + 1),	\
23*8e33eff8Schristos 		    ((t)1) << (i+1), "Unexpected result");		\
24*8e33eff8Schristos 	}								\
25*8e33eff8Schristos 									\
26*8e33eff8Schristos 	for (pow2 = 1; pow2 < 25; pow2++) {				\
27*8e33eff8Schristos 		for (x = (((t)1) << (pow2-1)) + 1; x <= ((t)1) << pow2;	\
28*8e33eff8Schristos 		    x++) {						\
29*8e33eff8Schristos 			assert_##suf##_eq(pow2_ceil_##suf(x),		\
30*8e33eff8Schristos 			    ((t)1) << pow2,				\
31*8e33eff8Schristos 			    "Unexpected result, x=%"pri, x);		\
32*8e33eff8Schristos 		}							\
33*8e33eff8Schristos 	}								\
34*8e33eff8Schristos } while (0)
35*8e33eff8Schristos 
36*8e33eff8Schristos TEST_BEGIN(test_pow2_ceil_u64) {
37*8e33eff8Schristos 	TEST_POW2_CEIL(uint64_t, u64, FMTu64);
38*8e33eff8Schristos }
39*8e33eff8Schristos TEST_END
40*8e33eff8Schristos 
41*8e33eff8Schristos TEST_BEGIN(test_pow2_ceil_u32) {
42*8e33eff8Schristos 	TEST_POW2_CEIL(uint32_t, u32, FMTu32);
43*8e33eff8Schristos }
44*8e33eff8Schristos TEST_END
45*8e33eff8Schristos 
46*8e33eff8Schristos TEST_BEGIN(test_pow2_ceil_zu) {
47*8e33eff8Schristos 	TEST_POW2_CEIL(size_t, zu, "zu");
48*8e33eff8Schristos }
49*8e33eff8Schristos TEST_END
50*8e33eff8Schristos 
51*8e33eff8Schristos int
52*8e33eff8Schristos main(void) {
53*8e33eff8Schristos 	return test(
54*8e33eff8Schristos 	    test_pow2_ceil_u64,
55*8e33eff8Schristos 	    test_pow2_ceil_u32,
56*8e33eff8Schristos 	    test_pow2_ceil_zu);
57*8e33eff8Schristos }
58