commit b90ac9e25cf3178bbb79f313478c89ab09080abc
author Tony Mak <tonymak@google.com> Wed Oct 28 15:32:18 2020 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Wed Oct 28 15:32:18 2020 +0000
tree 12199546210d2e899fb0f12d83c2d4a539681d53
parent a8f7845ef08c3a32d9bef96c1935e57a13822263
parent 4c0e6c69f87764a31e506a32546dbe9298c80338

Merge "Import abseil to external/libtextclassifier"

java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java

diff --git a/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java b/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java
index 6d01a64..30a9dca 100644
--- a/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java
+++ b/java/tests/instrumentation/src/com/android/textclassifier/subjects/EntitySubject.java
@@ -20,7 +20,6 @@
 
 import com.android.textclassifier.Entity;
 import com.google.common.truth.FailureMetadata;
-import com.google.common.truth.MathUtil;
 import com.google.common.truth.Subject;
 import javax.annotation.Nullable;
 
@@ -44,8 +43,7 @@
     if (!entity.getEntityType().equals(this.entity.getEntityType())) {
       failWithActual("expected to have type", entity.getEntityType());
     }
-    if (!MathUtil.equalWithinTolerance(entity.getScore(), this.entity.getScore(), TOLERANCE)) {
-      failWithActual("expected to have confidence score", entity.getScore());
-    }
+    check("expected to have confidence score").that(entity.getScore()).isWithin(TOLERANCE)
+      .of(this.entity.getScore());
   }
 }