Allow some tolerance in equality test by using EXPECT_FLOAT_EQ

galeone · Mar 16, 2022 · bef4a90 · bef4a90
1 parent da8aed5
commit bef4a90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/core/framework/model_test.cc b/tensorflow/core/framework/model_test.cc
@@ -459,7 +459,7 @@ TEST_P(AsyncUnknownRatioTest, Model) {
   async_unknown_many->record_element();
   // Estimated ratio is 2/3
   ratio = 2.0 / 3.0;
-  EXPECT_EQ(
+  EXPECT_FLOAT_EQ(
       async_unknown_many->TotalProcessingTime(/*processing_times=*/nullptr),
       ratio * (50 + 50) + 128 / 3.0);
   EXPECT_LE(async_unknown_many->OutputTime(&input_times, nullptr),