Skip to content

Commit 15b8ed6

Browse files
fixed github issue 3821 for string shannon entropy test (#3824)
* fixed github issue 3821 for string shannon entropy test * ahrav is legend
1 parent e98dfa5 commit 15b8ed6

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

pkg/detectors/falsepositives_test.go

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -154,14 +154,14 @@ func TestStringShannonEntropy(t *testing.T) {
154154
args: args{
155155
input: "aaaaaaaaaaaaaaaaaaaaaaaaaaab",
156156
},
157-
want: 0.22228483068568816,
157+
want: 0.22,
158158
},
159159
{
160160
name: "entropy 3",
161161
args: args{
162162
input: "aaaaaaaaaaaaaaaaaaaaaaaaaaabaaaaaaaaaaaaaaaaaaaaaaaaaaab",
163163
},
164-
want: 0.22228483068568816,
164+
want: 0.22,
165165
},
166166
{
167167
name: "empty",
@@ -173,8 +173,11 @@ func TestStringShannonEntropy(t *testing.T) {
173173
}
174174
for _, tt := range tests {
175175
t.Run(tt.name, func(t *testing.T) {
176-
if got := StringShannonEntropy(tt.args.input); got != tt.want {
177-
t.Errorf("StringShannonEntropy() = %v, want %v", got, tt.want)
176+
got := StringShannonEntropy(tt.args.input)
177+
if len(tt.args.input) > 0 && tt.want != 0 {
178+
assert.InEpsilon(t, tt.want, got, 0.1)
179+
} else {
180+
assert.Equal(t, tt.want, got)
178181
}
179182
})
180183
}

0 commit comments

Comments
 (0)