update SEH test to round (aws#733)

jefchien · Apr 10, 2023 · e28b76f · e28b76f
1 parent 32d78d1
commit e28b76f
Showing 1 changed file with 15 additions and 9 deletions.
diff --git a/metric/distribution/seh1/seh1_distribution_test.go b/metric/distribution/seh1/seh1_distribution_test.go
@@ -4,6 +4,7 @@
 package seh1
 
 import (
+	"math/big"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -24,11 +25,12 @@ func TestSEH1Distribution(t *testing.T) {
 	assert.Equal(t, "Count", dist.Unit())
 	values, counts := dist.ValuesAndCounts()
 	assert.Equal(t, len(values), len(counts))
-	valuesCountsMap := map[float64]float64{}
+	valuesCountsMap := map[string]float64{}
 	for i := 0; i < len(values); i++ {
-		valuesCountsMap[values[i]] = counts[i]
+		valuesCountsMap[truncate(values[i])] = counts[i]
 	}
-	expectedValuesCountsMap := map[float64]float64{20.13119624437727: 1, 29.474084421392764: 1, 52.21513847164702: 1}
+	expectedValuesCountsMap := map[string]float64{"20.13119624": 1, "29.47408442": 1, "52.21513847": 1}
+
 	assert.Equal(t, expectedValuesCountsMap, valuesCountsMap)
 
 	//another dist new and add entry
@@ -45,11 +47,11 @@ func TestSEH1Distribution(t *testing.T) {
 	assert.Equal(t, "", anotherDist.Unit())
 	values, counts = anotherDist.ValuesAndCounts()
 	assert.Equal(t, len(values), len(counts))
-	valuesCountsMap = map[float64]float64{}
+	valuesCountsMap = map[string]float64{}
 	for i := 0; i < len(values); i++ {
-		valuesCountsMap[values[i]] = counts[i]
+		valuesCountsMap[truncate(values[i])] = counts[i]
 	}
-	expectedValuesCountsMap = map[float64]float64{20.13119624437727: 1, 22.144315868814992: 3}
+	expectedValuesCountsMap = map[string]float64{"20.13119624": 1, "22.14431587": 3}
 	assert.Equal(t, expectedValuesCountsMap, valuesCountsMap)
 
 	//clone dist and anotherDist
@@ -65,11 +67,11 @@ func TestSEH1Distribution(t *testing.T) {
 	assert.Equal(t, "Count", dist.Unit())
 	values, counts = dist.ValuesAndCounts()
 	assert.Equal(t, len(values), len(counts))
-	valuesCountsMap = map[float64]float64{}
+	valuesCountsMap = map[string]float64{}
 	for i := 0; i < len(values); i++ {
-		valuesCountsMap[values[i]] = counts[i]
+		valuesCountsMap[truncate(values[i])] = counts[i]
 	}
-	expectedValuesCountsMap = map[float64]float64{20.13119624437727: 2, 22.144315868814992: 3, 29.474084421392764: 1, 52.21513847164702: 1}
+	expectedValuesCountsMap = map[string]float64{"20.13119624": 2, "22.14431587": 3, "29.47408442": 1, "52.21513847": 1}
 	assert.Equal(t, expectedValuesCountsMap, valuesCountsMap)
 
 	//add distClone into another dist
@@ -91,3 +93,7 @@ func cloneSEH1Distribution(dist *SEH1Distribution) *SEH1Distribution {
 	}
 	return clonedDist
 }
+
+func truncate(f float64) string {
+	return big.NewFloat(f).SetPrec(100).String()
+}