OPENNLP-1481: Adding delta for DL test assertions. (#521)

apache · Mar 15, 2023 · 8a942d3 · 8a942d3
1 parent cf04094
commit 8a942d3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/opennlp-dl/src/test/java/opennlp/dl/doccat/DocumentCategorizerDLEval.java b/opennlp-dl/src/test/java/opennlp/dl/doccat/DocumentCategorizerDLEval.java
@@ -83,7 +83,7 @@ public void categorize() throws IOException, OrtException {
     logger.debug("Actual: {}", Arrays.toString(sortedResult));
     logger.debug("Expected: {}", Arrays.toString(expected));
 
-    Assertions.assertArrayEquals(expected, sortedResult, 0.0);
+    Assertions.assertArrayEquals(expected, sortedResult, 0.000001);
     Assertions.assertEquals(5, result.length);
 
     final String category = documentCategorizerDL.getBestCategory(result);
@@ -119,7 +119,7 @@ public void categorizeWithGpu() throws Exception {
             0.3003573715686798,
             0.6352779865264893};
 
-    Assertions.assertArrayEquals(expected, result, 0.0);
+    Assertions.assertArrayEquals(expected, result, 0.000001);
     Assertions.assertEquals(5, result.length);
 
     final String category = documentCategorizerDL.getBestCategory(result);
@@ -151,7 +151,7 @@ public void categorizeWithInferenceOptions() throws Exception {
 
     final double[] expected = new double[]{0.8851314783096313, 0.11486853659152985};
 
-    Assertions.assertArrayEquals(expected, result, 0.0);
+    Assertions.assertArrayEquals(expected, result, 0.000001);
     Assertions.assertEquals(2, result.length);
 
     final String category = documentCategorizerDL.getBestCategory(result);
@@ -174,11 +174,11 @@ public void scoreMap() throws Exception {
 
     final Map<String, Double> result = documentCategorizerDL.scoreMap(new String[]{"I am happy"});
 
-    Assertions.assertEquals(0.6352779865264893, result.get("very good").doubleValue(), 0);
-    Assertions.assertEquals(0.3003573715686798, result.get("good").doubleValue(), 0);
-    Assertions.assertEquals(0.04995147883892059, result.get("neutral").doubleValue(), 0);
-    Assertions.assertEquals(0.006593209225684404, result.get("bad").doubleValue(), 0);
-    Assertions.assertEquals(0.007819971069693565, result.get("very bad").doubleValue(), 0);
+    Assertions.assertEquals(0.6352779865264893, result.get("very good").doubleValue(), 0.000001);
+    Assertions.assertEquals(0.3003573715686798, result.get("good").doubleValue(), 0.000001);
+    Assertions.assertEquals(0.04995147883892059, result.get("neutral").doubleValue(), 0.000001);
+    Assertions.assertEquals(0.006593209225684404, result.get("bad").doubleValue(), 0.000001);
+    Assertions.assertEquals(0.007819971069693565, result.get("very bad").doubleValue(), 0.000001);
 
   }
 

diff --git a/opennlp-dl/src/test/java/opennlp/dl/namefinder/NameFinderDLEval.java b/opennlp-dl/src/test/java/opennlp/dl/namefinder/NameFinderDLEval.java
@@ -64,7 +64,7 @@ public void tokenNameFinder1Test() throws Exception {
     Assertions.assertEquals(1, spans.length);
     Assertions.assertEquals(0, spans[0].getStart());
     Assertions.assertEquals(17, spans[0].getEnd());
-    Assertions.assertEquals(8.251646041870117, spans[0].getProb(), 0.0);
+    Assertions.assertEquals(8.251646041870117, spans[0].getProb(), 0.000001);
     Assertions.assertEquals("George Washington", spans[0].getCoveredText(String.join(" ", tokens)));
 
   }