diff --git a/core/mathx/entropy_test.go b/core/mathx/entropy_test.go index 3b6c2659..03db018d 100644 --- a/core/mathx/entropy_test.go +++ b/core/mathx/entropy_test.go @@ -15,3 +15,17 @@ func TestCalcEntropy(t *testing.T) { } assert.True(t, CalcEntropy(m) > .99) } + +func TestCalcEmptyEntropy(t *testing.T) { + m := make(map[any]int) + assert.Equal(t, float64(1), CalcEntropy(m)) +} + +func TestCalcDiffEntropy(t *testing.T) { + const total = 1000 + m := make(map[any]int, total) + for i := 0; i < total; i++ { + m[i] = i + } + assert.True(t, CalcEntropy(m) < .99) +} diff --git a/rest/handler/timeouthandler_test.go b/rest/handler/timeouthandler_test.go index ee6fdb91..f4f97d45 100644 --- a/rest/handler/timeouthandler_test.go +++ b/rest/handler/timeouthandler_test.go @@ -172,6 +172,7 @@ func TestTimeoutSSE(t *testing.T) { handler.ServeHTTP(resp, req) assert.Equal(t, http.StatusOK, resp.Code) } + func TestTimeoutWebsocket(t *testing.T) { timeoutHandler := TimeoutHandler(time.Millisecond) handler := timeoutHandler(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {