add failure logic for delta +Inf

gavv · May 3, 2023 · e13d9a5 · e13d9a5
1 parent 600d8e0
commit e13d9a5
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/number.go b/number.go
@@ -311,6 +311,16 @@ func (n *Number) InDeltaRelative(value, delta float64) *Number {
 		return n
 	}
 
+	if math.IsInf(delta, 0) {
+		opChain.fail(AssertionFailure{
+			Type: AssertUsage,
+			Errors: []error{
+				errors.New("unexpected Inf delta argument"),
+			},
+		})
+		return n
+	}
+
 	if delta < 0 {
 		opChain.fail(AssertionFailure{
 			Type: AssertUsage,
@@ -402,6 +412,16 @@ func (n *Number) NotInDeltaRelative(value, delta float64) *Number {
 		return n
 	}
 
+	if math.IsInf(delta, 0) {
+		opChain.fail(AssertionFailure{
+			Type: AssertUsage,
+			Errors: []error{
+				errors.New("unexpected Inf delta argument"),
+			},
+		})
+		return n
+	}
+
 	if delta < 0 {
 		opChain.fail(AssertionFailure{
 			Type: AssertUsage,

diff --git a/number_test.go b/number_test.go
@@ -393,11 +393,11 @@ func TestNumber_InDeltaRelative(t *testing.T) {
 			wantNotInDelta: failure,
 		},
 		{
-			name:           "+Inf delta in range",
+			name:           "delta is +Inf",
 			number:         1234.5,
 			value:          1234.0,
 			delta:          math.Inf(1),
-			wantInDelta:    success,
+			wantInDelta:    failure,
 			wantNotInDelta: failure,
 		},
 		{