diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-06-10 18:24:34 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-06-11 10:33:16 +0000 |
commit | 1625efc870f1a8b7c6e6382277ddbb245f91a294 (patch) | |
tree | 39fbbaa15ed7eb81337b082c2d20b0af68b91c02 /src/armnnSerializer/test/SerializerTestUtils.hpp | |
parent | 958e0ba61e940a8d11955cf2a10f681c7c47e1fa (diff) | |
download | armnn-1625efc870f1a8b7c6e6382277ddbb245f91a294.tar.gz |
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
Diffstat (limited to 'src/armnnSerializer/test/SerializerTestUtils.hpp')
-rw-r--r-- | src/armnnSerializer/test/SerializerTestUtils.hpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/src/armnnSerializer/test/SerializerTestUtils.hpp b/src/armnnSerializer/test/SerializerTestUtils.hpp index e085d2ef15..c6f148b1a1 100644 --- a/src/armnnSerializer/test/SerializerTestUtils.hpp +++ b/src/armnnSerializer/test/SerializerTestUtils.hpp @@ -12,8 +12,8 @@ #include <random> #include <vector> -#include <boost/test/unit_test.hpp> - +#include <cstdlib> +#include <doctest/doctest.h> armnn::INetworkPtr DeserializeNetwork(const std::string& serializerString); @@ -82,7 +82,7 @@ public: protected: void VerifyDescriptor(const Descriptor& descriptor) { - BOOST_CHECK(descriptor == m_Descriptor); + CHECK(descriptor == m_Descriptor); } Descriptor m_Descriptor; @@ -93,12 +93,12 @@ void CompareConstTensorData(const void* data1, const void* data2, unsigned int n { T typedData1 = static_cast<T>(data1); T typedData2 = static_cast<T>(data2); - BOOST_CHECK(typedData1); - BOOST_CHECK(typedData2); + CHECK(typedData1); + CHECK(typedData2); for (unsigned int i = 0; i < numElements; i++) { - BOOST_TEST(typedData1[i] == typedData2[i]); + CHECK(typedData1[i] == typedData2[i]); } } @@ -161,7 +161,7 @@ static std::vector<DataType> GenerateRandomData(size_t size) static std::default_random_engine generator; std::vector<DataType> randomData(size); - std::generate(randomData.begin(), randomData.end(), []() { return distribution(generator); }); + generate(randomData.begin(), randomData.end(), []() { return distribution(generator); }); return randomData; }
\ No newline at end of file |