diff options
Diffstat (limited to 'src/backends/reference/test/ArgMinMaxTests.cpp')
| -rw-r--r-- | src/backends/reference/test/ArgMinMaxTests.cpp | 22 |
1 files changed, 8 insertions, 14 deletions
diff --git a/src/backends/reference/test/ArgMinMaxTests.cpp b/src/backends/reference/test/ArgMinMaxTests.cpp index dce15b29ef..b79a108129 100644 --- a/src/backends/reference/test/ArgMinMaxTests.cpp +++ b/src/backends/reference/test/ArgMinMaxTests.cpp @@ -5,11 +5,11 @@ #include <reference/workloads/ArgMinMax.hpp> -#include <boost/test/unit_test.hpp> +#include <doctest/doctest.h> -BOOST_AUTO_TEST_SUITE(RefArgMinMax) - -BOOST_AUTO_TEST_CASE(ArgMinTest) +TEST_SUITE("RefArgMinMax") +{ +TEST_CASE("ArgMinTest") { const armnn::TensorInfo inputInfo({ 1, 2, 3 } , armnn::DataType::Float32); const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Signed64); @@ -25,14 +25,11 @@ BOOST_AUTO_TEST_CASE(ArgMinTest) armnn::ArgMinMaxFunction::Min, -2); - BOOST_CHECK_EQUAL_COLLECTIONS(outputValues.begin(), - outputValues.end(), - expectedValues.begin(), - expectedValues.end()); + CHECK(std::equal(outputValues.begin(), outputValues.end(), expectedValues.begin(), expectedValues.end())); } -BOOST_AUTO_TEST_CASE(ArgMaxTest) +TEST_CASE("ArgMaxTest") { const armnn::TensorInfo inputInfo({ 1, 2, 3 } , armnn::DataType::Float32); const armnn::TensorInfo outputInfo({ 1, 3 }, armnn::DataType::Signed64); @@ -48,11 +45,8 @@ BOOST_AUTO_TEST_CASE(ArgMaxTest) armnn::ArgMinMaxFunction::Max, -2); - BOOST_CHECK_EQUAL_COLLECTIONS(outputValues.begin(), - outputValues.end(), - expectedValues.begin(), - expectedValues.end()); + CHECK(std::equal(outputValues.begin(), outputValues.end(), expectedValues.begin(), expectedValues.end())); } -BOOST_AUTO_TEST_SUITE_END()
\ No newline at end of file +}
\ No newline at end of file |