[NeoML] CtcTest & MobileNetV2BlockTest add check (#984)

Signed-off-by: Kirill Golikov <kirill.golikov@abbyy.com>
neoml-lib · Oct 17, 2023 · 2f19b7b · 2f19b7b
1 parent 2877663
commit 2f19b7b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/NeoML/test/src/CtcTest.cpp b/NeoML/test/src/CtcTest.cpp
@@ -940,8 +940,8 @@ static void ctcTestImpl( const CTestParams& params, int seed )
 
 	EXPECT_TRUE( FloatEq( naiveLoss->GetLastLoss(), actualLoss->GetLastLoss(), 1e-4f ) ) << naiveLoss->GetLastLoss()
 		<< '\t' << actualLoss->GetLastLoss();
-	CompareBlobs( *naiveLoss->GetLastGradient(), *actualLoss->GetLastGradient(), 1e-4f );
-	CompareBlobs( *naiveLearn->ActualDiff, *actualLearn->ActualDiff, 1e-4f );
+	EXPECT_TRUE( CompareBlobs( *naiveLoss->GetLastGradient(), *actualLoss->GetLastGradient(), 1e-4f ) );
+	EXPECT_TRUE( CompareBlobs( *naiveLearn->ActualDiff, *actualLearn->ActualDiff, 1e-4f ) );
 }
 
 TEST_P( CCtcTest, Random )

diff --git a/NeoML/test/src/MobileNetV2BlockTest.cpp b/NeoML/test/src/MobileNetV2BlockTest.cpp
@@ -174,7 +174,7 @@ static void mobileNetV2BlockTestImpl( unsigned int seed, int freeTermMask, float
 
 	CPtr<CDnnBlob> expectedBlob = expectedSink->GetBlob();
 	CPtr<CDnnBlob> actualBlob = actualSink->GetBlob();
-	CompareBlobs( *expectedBlob, *actualBlob );
+	EXPECT_TRUE( CompareBlobs( *expectedBlob, *actualBlob, 1e-3 ) );
 }
 
 TEST( MobileNetV2BlockLayerTest, Run )