From 1625efc870f1a8b7c6e6382277ddbb245f91a294 Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Thu, 10 Jun 2021 18:24:34 +0100
Subject: IVGCVSW-5963 'Move unit tests to new framework'

* Used doctest in ArmNN unit tests

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ia9cf5fc72775878885c5f864abf2c56b3a935f1a
---
 src/armnn/test/FloatingPointConverterTest.cpp | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

(limited to 'src/armnn/test/FloatingPointConverterTest.cpp')

diff --git a/src/armnn/test/FloatingPointConverterTest.cpp b/src/armnn/test/FloatingPointConverterTest.cpp
index d3474de831..21a16a3cc0 100644
--- a/src/armnn/test/FloatingPointConverterTest.cpp
+++ b/src/armnn/test/FloatingPointConverterTest.cpp
@@ -8,11 +8,13 @@
 #include <BFloat16.hpp>
 #include <Half.hpp>
 
-#include <boost/test/unit_test.hpp>
+#include <vector>
 
-BOOST_AUTO_TEST_SUITE(TestFPConversion)
+#include <doctest/doctest.h>
 
-BOOST_AUTO_TEST_CASE(TestConvertFp32ToFp16)
+TEST_SUITE("TestFPConversion")
+{
+TEST_CASE("TestConvertFp32ToFp16")
 {
     using namespace half_float::literal;
 
@@ -27,14 +29,14 @@ BOOST_AUTO_TEST_CASE(TestConvertFp32ToFp16)
     {
         armnn::Half expected(floatArray[i]);
         armnn::Half actual = convertedBuffer[i];
-        BOOST_CHECK_EQUAL(expected, actual);
+        CHECK_EQ(expected, actual);
 
         float convertedHalf = actual;
-        BOOST_CHECK_CLOSE(floatArray[i], convertedHalf, 0.07);
+        CHECK_EQ(floatArray[i], doctest::Approx(convertedHalf).epsilon(0.07));
     }
 }
 
-BOOST_AUTO_TEST_CASE(TestConvertFp16ToFp32)
+TEST_CASE("TestConvertFp16ToFp32")
 {
     using namespace half_float::literal;
 
@@ -49,11 +51,11 @@ BOOST_AUTO_TEST_CASE(TestConvertFp16ToFp32)
     {
         float expected(halfArray[i]);
         float actual = convertedBuffer[i];
-        BOOST_CHECK_EQUAL(expected, actual);
+        CHECK_EQ(expected, actual);
     }
 }
 
-BOOST_AUTO_TEST_CASE(TestConvertFloat32ToBFloat16)
+TEST_CASE("TestConvertFloat32ToBFloat16")
 {
     float floatArray[] = { 1.704735E38f,   // 0x7F004000 round down
                            0.0f,           // 0x00000000 round down
@@ -102,11 +104,11 @@ BOOST_AUTO_TEST_CASE(TestConvertFloat32ToBFloat16)
     for (size_t i = 0; i < numFloats; i++)
     {
         armnn::BFloat16 actual = convertedBuffer[i];
-        BOOST_CHECK_EQUAL(expectedResult[i], actual.Val());
+        CHECK_EQ(expectedResult[i], actual.Val());
     }
 }
 
-BOOST_AUTO_TEST_CASE(TestConvertBFloat16ToFloat32)
+TEST_CASE("TestConvertBFloat16ToFloat32")
 {
     uint16_t bf16Array[] = { 16256, 16320, 38699, 16384, 49156, 32639 };
     size_t numFloats = sizeof(bf16Array) / sizeof(bf16Array[0]);
@@ -118,8 +120,8 @@ BOOST_AUTO_TEST_CASE(TestConvertBFloat16ToFloat32)
     for (size_t i = 0; i < numFloats; i++)
     {
         float actual = convertedBuffer[i];
-        BOOST_CHECK_EQUAL(expectedResult[i], actual);
+        CHECK_EQ(expectedResult[i], actual);
     }
 }
 
-BOOST_AUTO_TEST_SUITE_END()
+}
-- 
cgit v1.2.1