doc values assertions

Signed-off-by: Sarthak Aggarwal <[email protected]>
sarthakaggarwal97 · Aug 15, 2024 · 08b2ffe · 08b2ffe
1 parent e890b78
commit 08b2ffe
Show file tree

Hide file tree

Showing 3 changed files with 549 additions and 126 deletions.
diff --git a/...ain/java/org/opensearch/index/codec/composite/composite99/Composite99DocValuesReader.java b/...ain/java/org/opensearch/index/codec/composite/composite99/Composite99DocValuesReader.java
@@ -270,7 +270,6 @@ public CompositeIndexValues getCompositeIndexValues(CompositeIndexFieldInfo comp
 
         switch (compositeIndexFieldInfo.getType()) {
             case STAR_TREE:
-
                 return new StarTreeValues(
                     compositeIndexMetadataMap.get(compositeIndexFieldInfo.getField()),
                     compositeIndexInputMap.get(compositeIndexFieldInfo.getField()),

diff --git a/...rc/test/java/org/opensearch/index/compositeindex/datacube/startree/StarTreeTestUtils.java b/...rc/test/java/org/opensearch/index/compositeindex/datacube/startree/StarTreeTestUtils.java
@@ -125,7 +125,11 @@ public static void assertStarTreeDocuments(StarTreeDocument[] starTreeDocuments,
             }
 
             for (int mi = 0; mi < resultStarTreeDocument.metrics.length; mi++) {
-                assertEquals(resultStarTreeDocument.metrics[mi], expectedStarTreeDocument.metrics[mi]);
+                if (expectedStarTreeDocument.metrics[mi] instanceof Long) {
+                    assertEquals(resultStarTreeDocument.metrics[mi], ((Long) expectedStarTreeDocument.metrics[mi]).doubleValue());
+                } else {
+                    assertEquals(resultStarTreeDocument.metrics[mi], expectedStarTreeDocument.metrics[mi]);
+                }
             }
         }
     }