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/armnn/test/ModelAccuracyCheckerTest.cpp | |
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/armnn/test/ModelAccuracyCheckerTest.cpp')
-rw-r--r-- | src/armnn/test/ModelAccuracyCheckerTest.cpp | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/src/armnn/test/ModelAccuracyCheckerTest.cpp b/src/armnn/test/ModelAccuracyCheckerTest.cpp index 93dba7e75e..47f112ee72 100644 --- a/src/armnn/test/ModelAccuracyCheckerTest.cpp +++ b/src/armnn/test/ModelAccuracyCheckerTest.cpp @@ -4,7 +4,7 @@ // #include "ModelAccuracyChecker.hpp" -#include <boost/test/unit_test.hpp> +#include <doctest/doctest.h> #include <iostream> #include <string> @@ -50,12 +50,12 @@ struct TestHelper }; } -BOOST_AUTO_TEST_SUITE(ModelAccuracyCheckerTest) - +TEST_SUITE("ModelAccuracyCheckerTest") +{ using TContainer = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>, std::vector<int8_t>>; -BOOST_FIXTURE_TEST_CASE(TestFloat32OutputTensorAccuracy, TestHelper) +TEST_CASE_FIXTURE(TestHelper, "TestFloat32OutputTensorAccuracy") { ModelAccuracyChecker checker(GetValidationLabelSet(), GetModelOutputLabels()); @@ -70,7 +70,7 @@ BOOST_FIXTURE_TEST_CASE(TestFloat32OutputTensorAccuracy, TestHelper) // Top 1 Accuracy float totalAccuracy = checker.GetAccuracy(1); - BOOST_CHECK(totalAccuracy == 100.0f); + CHECK(totalAccuracy == 100.0f); // Add image 2 and check accuracy std::vector<float> inferenceOutputVector2 = {0.10f, 0.0f, 0.0f, 0.0f, 0.05f, 0.70f, 0.0f, 0.0f, 0.0f, 0.15f}; @@ -83,11 +83,11 @@ BOOST_FIXTURE_TEST_CASE(TestFloat32OutputTensorAccuracy, TestHelper) // Top 1 Accuracy totalAccuracy = checker.GetAccuracy(1); - BOOST_CHECK(totalAccuracy == 50.0f); + CHECK(totalAccuracy == 50.0f); // Top 2 Accuracy totalAccuracy = checker.GetAccuracy(2); - BOOST_CHECK(totalAccuracy == 100.0f); + CHECK(totalAccuracy == 100.0f); // Add image 3 and check accuracy std::vector<float> inferenceOutputVector3 = {0.0f, 0.10f, 0.0f, 0.0f, 0.05f, 0.70f, 0.0f, 0.0f, 0.0f, 0.15f}; @@ -100,15 +100,15 @@ BOOST_FIXTURE_TEST_CASE(TestFloat32OutputTensorAccuracy, TestHelper) // Top 1 Accuracy totalAccuracy = checker.GetAccuracy(1); - BOOST_CHECK(totalAccuracy == 33.3333321f); + CHECK(totalAccuracy == 33.3333321f); // Top 2 Accuracy totalAccuracy = checker.GetAccuracy(2); - BOOST_CHECK(totalAccuracy == 66.6666641f); + CHECK(totalAccuracy == 66.6666641f); // Top 3 Accuracy totalAccuracy = checker.GetAccuracy(3); - BOOST_CHECK(totalAccuracy == 100.0f); + CHECK(totalAccuracy == 100.0f); } -BOOST_AUTO_TEST_SUITE_END() +} |