Fix float64 comparison in stats_helpers_test.go

Signed-off-by: Akihiro Suda <[email protected]>

cli/command/container/stats_helpers_test.go

@@ -5,16 +5,22 @@ import (
 	"testing"
 )
 
+func float64Equals(x, y float64) bool {
+	epsilon := 1e-9
+	return x-y < epsilon && y-x < epsilon
+}
+
 func TestCalculateMemUsageUnixNoCache(t *testing.T) {
 	// Given
 	stats := types.MemoryStats{Usage: 500, Stats: map[string]uint64{"cache": 400}}
 
 	// When
 	result := calculateMemUsageUnixNoCache(stats)
+	expected := float64(100.0)
 
 	// Then
-	if result != 100 {
-		t.Errorf("mem = %d, want 100", result)
+	if !float64Equals(result, expected) {
+		t.Errorf("mem = %f, want %f", result, expected)
 	}
 }
 
@@ -28,14 +34,14 @@ func TestCalculateMemPercentUnixNoCache(t *testing.T) {
 	t.Run("Limit is set", func(t *testing.T) {
 		result := calculateMemPercentUnixNoCache(someLimit, used)
 		expected := float64(70.0)
-		if result != expected {
+		if !float64Equals(result, expected) {
 			t.Errorf("percent = %f, want %f", result, expected)
 		}
 	})
 	t.Run("No limit, no cgroup data", func(t *testing.T) {
 		result := calculateMemPercentUnixNoCache(noLimit, used)
 		expected := float64(0.0)
-		if result != expected {
+		if !float64Equals(result, expected) {
 			t.Errorf("percent = %f, want %f", result, expected)
 		}
 	})