1 // 2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "TestNameOnlyLayerVisitor.hpp" 7 8 #include <Network.hpp> 9 10 #include <boost/test/unit_test.hpp> 11 12 namespace 13 { 14 15 #define TEST_CASE_CHECK_LAYER_VISITOR_NAME(name) \ 16 BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorName) \ 17 { \ 18 Test##name##LayerVisitor visitor("name##Layer"); \ 19 armnn::Network net; \ 20 armnn::IConnectableLayer *const layer = net.Add##name##Layer("name##Layer"); \ 21 layer->Accept(visitor); \ 22 } 23 24 #define TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name) \ 25 BOOST_AUTO_TEST_CASE(Check##name##LayerVisitorNameNullptr) \ 26 { \ 27 Test##name##LayerVisitor visitor; \ 28 armnn::Network net; \ 29 armnn::IConnectableLayer *const layer = net.Add##name##Layer(); \ 30 layer->Accept(visitor); \ 31 } 32 33 #define TEST_SUITE_NAME_ONLY_LAYER_VISITOR(name) \ 34 TEST_CASE_CHECK_LAYER_VISITOR_NAME(name) \ 35 TEST_CASE_CHECK_LAYER_VISITOR_NAME_NULLPTR(name) 36 37 } // anonymous namespace 38 39 BOOST_AUTO_TEST_SUITE(TestNameOnlyLayerVisitor) 40 41 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Addition) 42 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Dequantize) 43 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Division) 44 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Floor) 45 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Maximum) 46 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Merge) 47 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Minimum) 48 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Multiplication) 49 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Prelu) 50 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Quantize) 51 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Rank) 52 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Subtraction) 53 TEST_SUITE_NAME_ONLY_LAYER_VISITOR(Switch) 54 55 BOOST_AUTO_TEST_SUITE_END() 56