From ef57900b163f64429fa82a542fe5662e3b41ef1e Mon Sep 17 00:00:00 2001
From: bizy <bizywiz@gmail.com>
Date: Tue, 5 Mar 2024 01:11:10 +0700
Subject: [PATCH] Add bench and tests for `runtime.Equal`

---
 vm/runtime/helpers_test.go | 57 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 vm/runtime/helpers_test.go

diff --git a/vm/runtime/helpers_test.go b/vm/runtime/helpers_test.go
new file mode 100644
index 00000000..42a0aece
--- /dev/null
+++ b/vm/runtime/helpers_test.go
@@ -0,0 +1,57 @@
+package runtime_test
+
+import (
+	"testing"
+
+	"github.com/expr-lang/expr/vm/runtime"
+	"github.com/stretchr/testify/assert"
+)
+
+var tests = []struct {
+	name string
+	a, b any
+	want bool
+}{
+	{"int == int", 42, 42, true},
+	{"int != int", 42, 33, false},
+	{"int == int8", 42, int8(42), true},
+	{"int == int16", 42, int16(42), true},
+	{"int == int32", 42, int32(42), true},
+	{"int == int64", 42, int64(42), true},
+	{"float == float", 42.0, 42.0, true},
+	{"float != float", 42.0, 33.0, false},
+	{"float == int", 42.0, 42, true},
+	{"float != int", 42.0, 33, false},
+	{"string == string", "foo", "foo", true},
+	{"string != string", "foo", "bar", false},
+	{"bool == bool", true, true, true},
+	{"bool != bool", true, false, false},
+	{"[]any == []int", []any{1, 2, 3}, []int{1, 2, 3}, true},
+	{"[]any != []int", []any{1, 2, 3}, []int{1, 2, 99}, false},
+	{"deep []any == []any", []any{[]int{1}, 2, []any{"3"}}, []any{[]any{1}, 2, []string{"3"}}, true},
+	{"deep []any != []any", []any{[]int{1}, 2, []any{"3", "42"}}, []any{[]any{1}, 2, []string{"3"}}, false},
+	{"map[string]any == map[string]any", map[string]any{"a": 1}, map[string]any{"a": 1}, true},
+	{"map[string]any != map[string]any", map[string]any{"a": 1}, map[string]any{"a": 1, "b": 2}, false},
+}
+
+func TestEqual(t *testing.T) {
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := runtime.Equal(tt.a, tt.b)
+			assert.Equal(t, tt.want, got, "Equal(%v, %v) = %v; want %v", tt.a, tt.b, got, tt.want)
+			got = runtime.Equal(tt.b, tt.a)
+			assert.Equal(t, tt.want, got, "Equal(%v, %v) = %v; want %v", tt.b, tt.a, got, tt.want)
+		})
+	}
+
+}
+
+func BenchmarkEqual(b *testing.B) {
+	for _, tt := range tests {
+		b.Run(tt.name, func(b *testing.B) {
+			for i := 0; i < b.N; i++ {
+				runtime.Equal(tt.a, tt.b)
+			}
+		})
+	}
+}