diff options
Diffstat (limited to 'src/armnn/test/TestNameOnlyLayerVisitor.cpp')
| -rw-r--r-- | src/armnn/test/TestNameOnlyLayerVisitor.cpp | 61 |
1 files changed, 35 insertions, 26 deletions
diff --git a/src/armnn/test/TestNameOnlyLayerVisitor.cpp b/src/armnn/test/TestNameOnlyLayerVisitor.cpp index 971d7eeab7..00d65f8e76 100644 --- a/src/armnn/test/TestNameOnlyLayerVisitor.cpp +++ b/src/armnn/test/TestNameOnlyLayerVisitor.cpp @@ -7,13 +7,13 @@ #include <Network.hpp> -#include <boost/test/unit_test.hpp> +#include <doctest/doctest.h> namespace { -#define TEST_CASE_CHECK_LAYER_VISITOR_NAME(name) \ -BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorName) \ +#define TEST_CASE_CHECK_LAYER_VISITOR_NAME(name, testName) \ +TEST_CASE(#testName) \ { \ Test##name##LayerVisitor visitor("name##Layer"); \ armnn::NetworkImpl net; \ @@ -21,8 +21,8 @@ BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorName) \ layer->Accept(visitor); \ } -#define TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name) \ -BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorNameNullptr) \ +#define TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name, testName) \ +TEST_CASE(#testName) \ { \ Test##name##LayerVisitor visitor; \ armnn::NetworkImpl net; \ @@ -30,26 +30,35 @@ BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorNameNullptr) \ layer->Accept(visitor); \ } -#define TEST_SUITE_NAME_ONLY_LAYER_VISITOR(name) \ -TEST_CASE_CHECK_LAYER_VISITOR_NAME(name) \ -TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name) - } // anonymous namespace -BOOST_AUTO_TEST_SUITE(TestNameOnlyLayerVisitor) - -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Addition) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Dequantize) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Division) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Floor) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Maximum) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Merge) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Minimum) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Multiplication) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Prelu) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Quantize) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Rank) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Subtraction) -TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Switch) - -BOOST_AUTO_TEST_SUITE_END() +TEST_SUITE("TestNameOnlyLayerVisitor") +{ +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Addition, CheckAdditionLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Addition, CheckAdditionLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Dequantize, CheckDequantizeLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Dequantize, CheckDequantizeLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Division, CheckDivisionLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Division, CheckDivisionLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Floor, CheckFloorLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Floor, CheckFloorLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Maximum, CheckMaximumLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Maximum, CheckMaximumLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Merge, CheckMergeLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Merge, CheckMergeLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Minimum, CheckMinimumLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Minimum, CheckMinimumLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Multiplication, CheckMultiplicationLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Multiplication, CheckMultiplicationLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Prelu, CheckPreluLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Prelu, CheckPreluLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Quantize, CheckQuantizeLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Quantize, CheckQuantizeLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Rank, CheckRankLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Rank, CheckRankLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Subtraction, CheckSubtractionLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Subtraction, CheckSubtractionLayerVisitorNameNullptr) +TEST_CASE_CHECK_LAYER_VISITOR_NAME(Switch, CheckSwitchLayerVisitorName) +TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(Switch, CheckSwitchLayerVisitorNameNullptr) + +} |