From e9f5253e33daa1e4b4b8f03d1cad95507207b88d Mon Sep 17 00:00:00 2001 From: Jihoon Lee Date: Mon, 14 Dec 2020 13:53:51 +0000 Subject: [PATCH] [Test] Add macro to check if backbone is enabled When backbone is not enabled, test fails because backbone is not enabled This patch adds a define in the test so that test can pass when backbone is not enabled Signed-off-by: Jihoon Lee --- test/ccapi/unittest_ccapi.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/ccapi/unittest_ccapi.cpp b/test/ccapi/unittest_ccapi.cpp index 5912a78..efb70af 100644 --- a/test/ccapi/unittest_ccapi.cpp +++ b/test/ccapi/unittest_ccapi.cpp @@ -78,11 +78,15 @@ TEST(ccapi_layer, construct_02_p) { EXPECT_NO_THROW(layer = ml::train::layer::MultiOut()); EXPECT_EQ(layer->getType(), "output"); +#ifdef ENABLE_NNSTREAMER_BACKBONE EXPECT_NO_THROW(layer = ml::train::layer::BackboneNNStreamer()); EXPECT_EQ(layer->getType(), "backbone_nnstreamer"); +#endif +#ifdef ENABLE_TFLITE_BACKBONE EXPECT_NO_THROW(layer = ml::train::layer::BackboneTFLite()); EXPECT_EQ(layer->getType(), "backbone_tflite"); +#endif EXPECT_NO_THROW(layer = ml::train::layer::ReLU()); EXPECT_EQ(layer->getType(), "activation"); -- 2.7.4