util_test.go

package x

import (
	"testing"

	assert "github.com/stretchr/testify/assert"
)

func TestToInt(t *testing.T) {
	var o ObservationBits
	o = ObservationBits{false, true, true, false, true}
	assert.Equal(t, Observation(13), ToInt(o))
	o = ObservationBits{false, false, false}
	assert.Equal(t, Observation(0), ToInt(o))
	o = ObservationBits{true, false, false, false, false}
	assert.Equal(t, Observation(16), ToInt(o))
	o = ObservationBits{false, false, false, false, true}
	assert.Equal(t, Observation(1), ToInt(o))
}

func TestArgMax(t *testing.T) {
	var A []float64
	A = []float64{1.1, 3.3, -123.2, 40000.1, 1e6, -1e7, 22.3, -3.14}
	i, _ := ArgMax(A)
	assert.Equal(t, 4, i)
}

func TestLog2(t *testing.T) {
	assert.Equal(t, 3, Log2(8))
	assert.Equal(t, 5, Log2(32))
	assert.Equal(t, 11, Log2(2048))
	assert.Equal(t, 0, Log2(1))
	assert.Equal(t, 1, Log2(2))
	assert.Equal(t, 3, Log2(9))
	assert.Equal(t, 3, Log2(15))
}