From b08a58aab7c0600acf2f175e004ae9d3b250d90d Mon Sep 17 00:00:00 2001
From: troyyyyy <49091847+troyyyyy@users.noreply.github.com>
Date: Thu, 2 Mar 2023 09:24:06 +0800
Subject: [PATCH] Add check_equal

---
 utils/utils.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/utils/utils.py b/utils/utils.py
index f4db6f2..1138361 100644
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -72,3 +72,18 @@ def remapping_res(pred_res, m):
     for key, value in m.items():
         remapping[value] = key
     return [[remapping[symbol] for symbol in formula] for formula in pred_res]
+
+def check_equal(a, b):
+    if isinstance(a, (int, float)) and isinstance(b, (int, float)):
+        return abs(a - b) <= 1e-3
+    
+    if isinstance(a, list) and isinstance(b, list):
+        if len(a) != len(b):
+            return False
+        for i in range(len(a)):
+            if not check_equal(a[i], b[i]):
+                return False
+        return True
+    
+    else:    
+        return a == b