Fix flaky test

Signed-off-by: Lantao Jin <ltjin@amazon.com>

Author: LantaoJin

integ-test/src/test/java/org/opensearch/sql/calcite/tpch/CalcitePPLTpchIT.java

@@ -54,7 +54,7 @@ public void testQ1() throws IOException {
         schema("count_order", "bigint"));
     verifyDataRows(
         actual,
-        rows(
+        closeTo(
             "A",
             "F",
             37474,
@@ -65,7 +65,7 @@ public void testQ1() throws IOException {
             isPushdownEnabled() ? 25419.231826792962 : 25419.231826792948,
             isPushdownEnabled() ? 0.0508660351826793 : 0.050866035182679493,
             1478),
-        rows(
+        closeTo(
             "N",
             "F",
             1041,
@@ -76,7 +76,7 @@ public void testQ1() throws IOException {
             27402.659736842103,
             isPushdownEnabled() ? 0.04289473684210526 : 0.042894736842105284,
             38),
-        rows(
+        closeTo(
             "N",
             "O",
             75168,
@@ -87,7 +87,7 @@ public void testQ1() throws IOException {
             isPushdownEnabled() ? 25632.42277116627 : 25632.422771166166,
             isPushdownEnabled() ? 0.049697381842910573 : 0.04969738184291069,
             2941),
-        rows(
+        closeTo(
             "R",
             "F",
             36511,

integ-test/src/test/java/org/opensearch/sql/util/MatcherUtils.java

@@ -292,16 +292,21 @@ protected boolean matchesSafely(JSONArray array) {
     };
   }
 
-  public static TypeSafeMatcher<JSONArray> closeTo(Number... values) {
+  public static TypeSafeMatcher<JSONArray> closeTo(Object... values) {
     final double error = 1e-10;
     return new TypeSafeMatcher<JSONArray>() {
       @Override
       protected boolean matchesSafely(JSONArray item) {
-        List<Number> expectedValues = new ArrayList<>(Arrays.asList(values));
-        List<Number> actualValues = new ArrayList<>();
-        item.iterator().forEachRemaining(v -> actualValues.add((Number) v));
+        List<Object> expectedValues = new ArrayList<>(Arrays.asList(values));
+        List<Object> actualValues = new ArrayList<>();
+        item.iterator().forEachRemaining(v -> actualValues.add((Object) v));
         return actualValues.stream()
-            .allMatch(v -> valuesAreClose(v, expectedValues.get(actualValues.indexOf(v))));
+            .allMatch(
+                v ->
+                    v instanceof Number
+                        ? valuesAreClose(
+                            (Number) v, (Number) expectedValues.get(actualValues.indexOf(v)))
+                        : v.equals(expectedValues.get(actualValues.indexOf(v))));
       }
 
       @Override