diff options
Diffstat (limited to 'src/armnn/test/UnitTests.hpp')
-rw-r--r-- | src/armnn/test/UnitTests.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/test/UnitTests.hpp b/src/armnn/test/UnitTests.hpp index c15477bf19..b55b13d4c8 100644 --- a/src/armnn/test/UnitTests.hpp +++ b/src/armnn/test/UnitTests.hpp @@ -42,7 +42,8 @@ void CompareTestResultIfSupported(const std::string& testName, const LayerTestRe "The test name does not match the supportedness it is reporting"); if (testResult.supported) { - BOOST_TEST(CompareTensors(testResult.output, testResult.outputExpected, testResult.compareBoolean)); + auto result = CompareTensors(testResult.output, testResult.outputExpected, testResult.compareBoolean); + BOOST_TEST(result.m_Result, result.m_Message.str()); } } @@ -56,7 +57,8 @@ void CompareTestResultIfSupported(const std::string& testName, const std::vector "The test name does not match the supportedness it is reporting"); if (testResult[i].supported) { - BOOST_TEST(CompareTensors(testResult[i].output, testResult[i].outputExpected)); + auto result = CompareTensors(testResult[i].output, testResult[i].outputExpected); + BOOST_TEST(result.m_Result, result.m_Message.str()); } } } |