erigon-pulse/cl/utils/math_test.go

package utils_test

import (
	"math"
	"testing"

	"github.com/ledgerwatch/erigon/cl/utils"
)

func TestIsPowerOf2(t *testing.T) {
	testCases := []struct {
		n          uint64
		isPowerOf2 bool
	}{
		{0, false},
		{1, true},
		{2, true},
		{3, false},
		{4, true},
		{16, true},
		{17, false},
		{64, true},
		{100, false},
		{256, true},
	}

	for _, tc := range testCases {
		isPowerOf2 := utils.IsPowerOf2(tc.n)
		if isPowerOf2 != tc.isPowerOf2 {
			t.Errorf("IsPowerOf2 returned incorrect result for %d. Expected: %t, Got: %t", tc.n, tc.isPowerOf2, isPowerOf2)
		}
	}
}

func TestPowerOf2(t *testing.T) {
	testCases := []struct {
		n            uint64
		expectedPow2 uint64
		shouldPanic  bool
	}{
		{0, 1, false},
		{1, 2, false},
		{2, 4, false},
		{3, 8, false},
		{4, 16, false},
		{63, 9223372036854775808, false},
		{64, 0, true},
	}

	for _, tc := range testCases {
		var pow2 uint64
		func() {
			defer func() {
				if r := recover(); r != nil {
					pow2 = 0
				}
			}()

			pow2 = utils.PowerOf2(tc.n)
		}()

		if pow2 != tc.expectedPow2 {
			t.Errorf("PowerOf2 returned incorrect result for %d. Expected: %d, Got: %d", tc.n, tc.expectedPow2, pow2)
		}

		if (pow2 == 0) != tc.shouldPanic {
			t.Errorf("PowerOf2 panicked unexpectedly for %d. Expected panic: %t", tc.n, tc.shouldPanic)
		}
	}
}

func TestIntegerSquareRoot(t *testing.T) {
	testCases := []struct {
		n            uint64
		expectedSqrt uint64
	}{
		{0, 0},
		{1, 1},
		{3, 1},
		{4, 2},
		{63, 7},
		{64, 8},
		{100, 10},
		{255, 15},
		{256, 16},
		{1000, 31},
		{1000000, 1000},
		{18446744073709551615, 4294967295}, // Maximum uint64
	}

	for _, tc := range testCases {
		sqrt := utils.IntegerSquareRoot(tc.n)
		if sqrt != tc.expectedSqrt {
			t.Errorf("IntegerSquareRoot returned incorrect result for %d. Expected: %d, Got: %d", tc.n, tc.expectedSqrt, sqrt)
		}
	}
}

func TestMax64(t *testing.T) {
	testCases := []struct {
		a        uint64
		b        uint64
		expected uint64
	}{
		{0, 0, 0},
		{0, 1, 1},
		{1, 0, 1},
		{1, 1, 1},
		{10, 5, 10},
		{5, 10, 10},
		{math.MaxUint64, 0, math.MaxUint64},
		{0, math.MaxUint64, math.MaxUint64},
		{math.MaxUint64, math.MaxUint64, math.MaxUint64},
	}

	for _, tc := range testCases {
		max := utils.Max64(tc.a, tc.b)
		if max != tc.expected {
			t.Errorf("Max64 returned incorrect result for %d and %d. Expected: %d, Got: %d", tc.a, tc.b, tc.expected, max)
		}
	}
}

func TestMin64(t *testing.T) {
	testCases := []struct {
		a        uint64
		b        uint64
		expected uint64
	}{
		{0, 0, 0},
		{0, 1, 0},
		{1, 0, 0},
		{1, 1, 1},
		{10, 5, 5},
		{5, 10, 5},
		{math.MaxUint64, 0, 0},
		{0, math.MaxUint64, 0},
		{math.MaxUint64, math.MaxUint64, math.MaxUint64},
	}

	for _, tc := range testCases {
		min := utils.Min64(tc.a, tc.b)
		if min != tc.expected {
			t.Errorf("Min64 returned incorrect result for %d and %d. Expected: %d, Got: %d", tc.a, tc.b, tc.expected, min)
		}
	}
}