commit a7df95289a7d0b8855c09829b191500dc2d8cfcb
author Alexander Dorokhine <adorokhine@google.com> Fri Oct 02 23:23:55 2020 -0700
committer Alexander Dorokhine <adorokhine@google.com> Tue Oct 06 00:14:52 2020 -0700
tree a315050214cb3efd083842131cadbe4f96e25838
parent 0cf39e4cdebe9277b2984817286f51780637d865

Migrate away from deprecated Truth APIs.

This is a transitional step towards truth 1.0.1, where these APIs have
been completely removed.

Bug: 168765701
Test: m checkbuild
Merged-In: I17f55bf97900c6aa0eba49624de8c723e431f6ed
Change-Id: I17f55bf97900c6aa0eba49624de8c723e431f6ed
(cherry picked from commit 06edbfae58fa80303b47bb062086ed9dca943f68)

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());
   }
 }